File tree 1 file changed +4
-4
lines changed
.circleci/unittest/linux/scripts
1 file changed +4
-4
lines changed Original file line number Diff line number Diff line change 22
22
echo " Using CUDA $CUDA_VERSION as determined by CU_VERSION: ${CU_VERSION} "
23
23
version=" $( python -c " print('.'.join(\" ${CUDA_VERSION} \" .split('.')[:2]))" ) "
24
24
cudatoolkit=" pytorch-cuda=${version} "
25
+
26
+ # make sure local cuda is set to required cuda version and not CUDA version by default
27
+ rm -f /usr/local/cuda
28
+ ln -s /usr/local/cuda-${version} /usr/local/cuda
25
29
fi
26
30
27
31
case " $( uname -s) " in
@@ -34,10 +38,6 @@ if [ "${os}" == "MacOSX" ]; then
34
38
conda install -y -c " pytorch-${UPLOAD_CHANNEL} " " pytorch-${UPLOAD_CHANNEL} " ::pytorch " ${cudatoolkit} "
35
39
else
36
40
conda install -y -c " pytorch-${UPLOAD_CHANNEL} " -c nvidia " pytorch-${UPLOAD_CHANNEL} " ::pytorch[build= " *${version} *" ] " ${cudatoolkit} "
37
-
38
- # make sure local cuda is set to required cuda version and not CUDA version by default
39
- rm -f /usr/local/cuda
40
- ln -s /usr/local/cuda-${version} /usr/local/cuda
41
41
fi
42
42
43
43
You can’t perform that action at this time.
0 commit comments