From cb73882b7f6b48f4ba73426b140e77d0d1d97468 Mon Sep 17 00:00:00 2001 From: sniklaus Date: Sat, 9 Sep 2017 22:59:59 -0700 Subject: no message --- install.bash | 10 ++++++++++ 1 file changed, 10 insertions(+) create mode 100644 install.bash (limited to 'install.bash') 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 -- cgit v1.2.3-70-g09d2