Skip to content

Commit 9ed39ad

Browse files
committed
Remove cpuonly from macOS
1 parent cf68fad commit 9ed39ad

File tree

1 file changed

+5
-2
lines changed

1 file changed

+5
-2
lines changed

.circleci/unittest/linux/scripts/install.sh

Lines changed: 5 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -11,13 +11,16 @@ eval "$(./conda/bin/conda shell.bash hook)"
1111
conda activate ./env
1212

1313
if [ -z "${CUDA_VERSION:-}" ] ; then
14-
cudatoolkit="cpuonly"
14+
case "$(uname -s)" in
15+
Darwin*) cudatoolkit="";;
16+
*) cudatoolkit="cpuonly"
17+
esac
1518
else
1619
version="$(python -c "print('.'.join(\"${CUDA_VERSION}\".split('.')[:2]))")"
1720
cudatoolkit="cudatoolkit=${version}"
1821
fi
1922
printf "Installing PyTorch with %s\n" "${cudatoolkit}"
20-
conda install -y -c "pytorch-${UPLOAD_CHANNEL}" pytorch "${cudatoolkit}"
23+
conda install -y -c "pytorch-${UPLOAD_CHANNEL}" pytorch ${cudatoolkit}
2124

2225
printf "* Installing torchaudio\n"
2326
BUILD_SOX=1 python setup.py install

0 commit comments

Comments
 (0)