Skip to content

Commit a49ba37

Browse files
committed
update cuda
1 parent b111a8c commit a49ba37

File tree

2 files changed

+3
-2
lines changed

2 files changed

+3
-2
lines changed

script/cuda.sh

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -83,6 +83,7 @@ if [ "${TRAVIS_OS_NAME}" = "windows" ] && [ "${IDX}" != "cpu" ]; then
8383
ls /c/Program\ Files/NVIDIA\ GPU\ Computing\ Toolkit
8484
ls /c/Program\ Files/NVIDIA\ GPU\ Computing\ Toolkit/CUDA
8585
ls /c/Program\ Files/NVIDIA\ GPU\ Computing\ Toolkit/CUDA/v${CUDA_SHORT}
86-
PATH=/c/Program\ Files/NVIDIA\ GPU\ Computing\ Toolkit/CUDA/v${CUDA_SHORT}/bin:$PATH
86+
CUDA_HOME=/c/Program\ Files/NVIDIA\ GPU\ Computing\ Toolkit/CUDA/v${CUDA_SHORT}
87+
PATH=${CUDA_HOME}/bin:$PATH
8788
nvcc --version
8889
fi

script/rename_wheel.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -14,6 +14,6 @@
1414
continue
1515

1616
names = name.split('-')
17-
name = '-'.join(names[:-5] + [names[-5] + '%2B' + idx] + names[-4:])
17+
name = '-'.join(names[:-4] + [idx] + names[-4:])
1818
new_wheel = osp.join(*wheel.split(osp.sep)[:-1], name)
1919
os.rename(wheel, new_wheel)

0 commit comments

Comments
 (0)