"data/once/git@developer.sourcefind.cn:OpenDAS/openpcdet.git" did not exist on "e948c537858d4e468a9ef50b023324f6bc96db2a"
Commit 9b139a9f authored by Ben Graham's avatar Ben Graham Committed by GitHub
Browse files

Update setup.py

parent 1fc78f87
......@@ -13,7 +13,7 @@ torch_dir = os.path.dirname(torch.__file__)
print('Building SCN module')
if torch.cuda.is_available():
r = os.system(
'cd sparseconvnet/SCN; nvcc init.cu -c -o init.cu.o -ccbin /usr/bin/cc -m64 --std c++11 -Xcompiler ,\"-fopenmp\",\"-fPIC\",\"-O3\",\"-DNDEBUG\" -gencode arch=compute_52,code=sm_52 -gencode arch=compute_20,code=sm_21 -DNVCC -I/usr/local/cuda/include -I' +
'cd sparseconvnet/SCN; nvcc init.cu -c -o init.cu.o -ccbin /usr/bin/cc -m64 --std c++11 -Xcompiler ,\"-fopenmp\",\"-fPIC\",\"-O3\",\"-DNDEBUG\" -gencode arch=compute_52,code=sm_52 -gencode arch=compute_30,code=sm_30 -DNVCC -I/usr/local/cuda/include -I' +
torch_dir +
'/lib/include -I' +
torch_dir +
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment