diff --git a/run_tests.sh b/run_tests.sh index b18d95192..57167ed5d 100755 --- a/run_tests.sh +++ b/run_tests.sh @@ -68,10 +68,10 @@ if [[ "$package_type" == conda || "$(uname)" == Darwin ]]; then if [[ "$cuda_ver" != 'cpu' ]]; then retry conda install -yq cudatoolkit=$cuda_ver_majmin else - # We DON'T want to install cpu-only, because it should not be + # We DON'T want to install cpuonly, because it should not be # necessary for OS X PyTorch which is always cpu only by default if [[ "$(uname)" != Darwin ]]; then - retry conda install -yq cpu-only -c pytorch + retry conda install -yq cpuonly -c pytorch fi fi retry conda install -yq cffi future hypothesis mkl>=2018 ninja numpy>=1.11 protobuf pytest setuptools six typing pyyaml