diff options
| author | sniklaus <simon.niklaus@outlook.com> | 2017-09-09 22:59:59 -0700 |
|---|---|---|
| committer | sniklaus <simon.niklaus@outlook.com> | 2017-09-09 22:59:59 -0700 |
| commit | cb73882b7f6b48f4ba73426b140e77d0d1d97468 (patch) | |
| tree | b2a45d643d3703e489ae2fd18ffd1143b4c7df3e /install.bash | |
no message
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 |