We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent cf68fad commit 9ed39adCopy full SHA for 9ed39ad
.circleci/unittest/linux/scripts/install.sh
@@ -11,13 +11,16 @@ eval "$(./conda/bin/conda shell.bash hook)"
11
conda activate ./env
12
13
if [ -z "${CUDA_VERSION:-}" ] ; then
14
- cudatoolkit="cpuonly"
+ case "$(uname -s)" in
15
+ Darwin*) cudatoolkit="";;
16
+ *) cudatoolkit="cpuonly"
17
+ esac
18
else
19
version="$(python -c "print('.'.join(\"${CUDA_VERSION}\".split('.')[:2]))")"
20
cudatoolkit="cudatoolkit=${version}"
21
fi
22
printf "Installing PyTorch with %s\n" "${cudatoolkit}"
-conda install -y -c "pytorch-${UPLOAD_CHANNEL}" pytorch "${cudatoolkit}"
23
+conda install -y -c "pytorch-${UPLOAD_CHANNEL}" pytorch ${cudatoolkit}
24
25
printf "* Installing torchaudio\n"
26
BUILD_SOX=1 python setup.py install
0 commit comments