fix facebookresearch#225 incompatibility with g++ 11

emepetres · Apr 5, 2022 · a6b5d13 · a6b5d13
1 parent dcf6a7f
commit a6b5d13
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 0 deletions.
diff --git a/build.sh b/build.sh
@@ -7,4 +7,6 @@
 
 #export TORCH_CUDA_ARCH_LIST="6.0;6.1;6.2;7.0;7.5"
 rm -rf build/ dist/ sparseconvnet.egg-info
+export CC=/usr/bin/gcc-10
+export CXX=/usr/bin/g++-10
 python setup.py install && python examples/hello-world.py
diff --git a/develop.sh b/develop.sh
@@ -7,4 +7,6 @@
 
 #export TORCH_CUDA_ARCH_LIST="6.0;6.1;6.2;7.0;7.5"
 rm -rf build/ dist/ sparseconvnet.egg-info sparseconvnet_SCN*.so
+export CC=/usr/bin/gcc-10
+export CXX=/usr/bin/g++-10
 python setup.py develop && python examples/hello-world.py