diff --git a/build.sh b/build.sh index 319346a..4a01802 100644 --- 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 index 21b4b4e..5537ee2 100644 --- 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