diff options
Diffstat (limited to 'install.bash')
| -rw-r--r-- | install.bash | 10 |
1 files changed, 10 insertions, 0 deletions
diff --git a/install.bash b/install.bash new file mode 100644 index 0000000..75c4b04 --- /dev/null +++ b/install.bash @@ -0,0 +1,10 @@ +#!/bin/bash + +TORCH=$(python -c "import os; import torch; print(os.path.dirname(torch.__file__))") + +nvcc -c -o src/SeparableConvolution_kernel.o src/SeparableConvolution_kernel.cu --gpu-architecture=compute_52 --gpu-code=compute_52 --compiler-options -fPIC -I ${TORCH}/lib/include/TH -I ${TORCH}/lib/include/THC + +python install.py + +wget --timestamping http://content.sniklaus.com/SepConv/network-l1.pytorch +wget --timestamping http://content.sniklaus.com/SepConv/network-lf.pytorch
\ No newline at end of file |