blob: 75c4b041acad2bc2097ea8ae1637ed536e0b687e (
plain)
1
2
3
4
5
6
7
8
9
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
|
