Skip to content

GitLab

Unverified Commit af048198 authored by Andrey Talman's avatar Andrey Talman Committed by GitHub
Browse files

Fix cpu instals in circleci (#7357)

parent cd332463
Show whitespace changes
Inline Side-by-side
Showing with 4 additions and 4 deletions
.circleci/unittest/linux/scripts/install.sh
View file @ af048198
...@@ -22,6 +22,10 @@ else ...@@ -22,6 +22,10 @@ else
echo "Using CUDA $CUDA_VERSION as determined by CU_VERSION: ${CU_VERSION} " echo "Using CUDA $CUDA_VERSION as determined by CU_VERSION: ${CU_VERSION} "
version="$(python -c "print('.'.join(\"${CUDA_VERSION}\".split('.')[:2]))")" version="$(python -c "print('.'.join(\"${CUDA_VERSION}\".split('.')[:2]))")"
cudatoolkit="pytorch-cuda=${version}" cudatoolkit="pytorch-cuda=${version}"
# make sure local cuda is set to required cuda version and not CUDA version by default
rm -f /usr/local/cuda
ln -s /usr/local/cuda-${version} /usr/local/cuda
fi fi
case "$(uname -s)" in case "$(uname -s)" in
...@@ -34,10 +38,6 @@ if [ "${os}" == "MacOSX" ]; then ...@@ -34,10 +38,6 @@ if [ "${os}" == "MacOSX" ]; then
conda install -y -c "pytorch-${UPLOAD_CHANNEL}" "pytorch-${UPLOAD_CHANNEL}"::pytorch "${cudatoolkit}" conda install -y -c "pytorch-${UPLOAD_CHANNEL}" "pytorch-${UPLOAD_CHANNEL}"::pytorch "${cudatoolkit}"
else else
conda install -y -c "pytorch-${UPLOAD_CHANNEL}" -c nvidia "pytorch-${UPLOAD_CHANNEL}"::pytorch[build="*${version}*"] "${cudatoolkit}" conda install -y -c "pytorch-${UPLOAD_CHANNEL}" -c nvidia "pytorch-${UPLOAD_CHANNEL}"::pytorch[build="*${version}*"] "${cudatoolkit}"
# make sure local cuda is set to required cuda version and not CUDA version by default
rm -f /usr/local/cuda
ln -s /usr/local/cuda-${version} /usr/local/cuda
fi fi
......
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