Skip to content

GitLab

Commit f9777244 authored by rusty1s's avatar rusty1s
Browse files

path

parent 56564cc5
Hide whitespace changes
Inline Side-by-side
Showing with 4 additions and 4 deletions
script/install.sh
View file @ f9777244
...@@ -20,7 +20,6 @@ if [ "${TRAVIS_OS_NAME}" = "linux" ] && [ "${IDX}" != "cpu" ]; then ...@@ -20,7 +20,6 @@ if [ "${TRAVIS_OS_NAME}" = "linux" ] && [ "${IDX}" != "cpu" ]; then
fi fi
if [ "${TRAVIS_OS_NAME}" = "windows" ] && [ "${IDX}" != "cpu" ]; then if [ "${TRAVIS_OS_NAME}" = "windows" ] && [ "${IDX}" != "cpu" ]; then
wget "https://developer.nvidia.com/compute/cuda/${CUDA_SHORT}/Prod2/local_installers2/cuda_${CUDA}_win10" wget -q "https://developer.nvidia.com/compute/cuda/${CUDA_SHORT}/Prod2/local_installers2/cuda_${CUDA}_win10"
ls ./cuda_${CUDA}_win10
./cuda_${CUDA}_win10.exe
fi fi
setup.py
View file @ f9777244
...@@ -10,7 +10,7 @@ from torch.utils.cpp_extension import CppExtension, CUDAExtension, CUDA_HOME ...@@ -10,7 +10,7 @@ from torch.utils.cpp_extension import CppExtension, CUDAExtension, CUDA_HOME
WITH_CUDA = torch.cuda.is_available() and CUDA_HOME is not None WITH_CUDA = torch.cuda.is_available() and CUDA_HOME is not None
if os.getenv('FORCE_CUDA', '0') == '1': if os.getenv('FORCE_CUDA', '0') == '1':
WITH_CUDA = True WITH_CUDA = True
if os.getenv('FORCE_NO_CUDA', '0') == '1': if os.getenv('FORCE_CPU', '0') == '1':
WITH_CUDA = False WITH_CUDA = False
BUILD_DOCS = os.getenv('BUILD_DOCS', '0') == '1' BUILD_DOCS = os.getenv('BUILD_DOCS', '0') == '1'
......
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