mirror of
https://github.com/wassname/Flex-Convolution.git
synced 2026-09-09 11:14:10 +08:00
12 lines
417 B
YAML
12 lines
417 B
YAML
pipeline:
|
|
custom_op:
|
|
image: tensorflow:ubuntu16.04-cuda9.2-bazel0.16.1-tensorflow1.10.0
|
|
environment:
|
|
- CUB_INC=/extra/cub-1.8.0/
|
|
commands:
|
|
- git clone --single-branch --depth 1 https://github.com/NVIDIA/cuda-samples.git /extra/samples
|
|
- export LD_LIBRARY_PATH=/usr/local/cuda/lib64/stubs:$${LD_LIBRARY_PATH}
|
|
- cd user_ops
|
|
- cmake . -DPYTHON_EXECUTABLE=python2
|
|
- make
|