Skip to content

Commit 37692d8

Browse files
authoredDec 21, 2020
[CCI] Install the CPU package for pytorch (#1105)
There's an issue with the CUDA 11.0 package for conda with conda-package-handling that relates to conda/conda-package-handling#71 This should fix issues with the conda smoke testing we encountered previously. Signed-off-by: Eli Uriegas <[email protected]>
1 parent e01724c commit 37692d8

File tree

2 files changed

+4
-4
lines changed

2 files changed

+4
-4
lines changed
 

‎.circleci/config.yml

+2-2
Original file line numberDiff line numberDiff line change
@@ -283,7 +283,7 @@ jobs:
283283
command: |
284284
set -x
285285
source /usr/local/etc/profile.d/conda.sh && conda activate python${PYTHON_VERSION}
286-
conda install -v -y -c pytorch-${UPLOAD_CHANNEL} pytorch
286+
conda install -v -y -c pytorch-${UPLOAD_CHANNEL} pytorch cpuonly
287287
conda install -v -y -c file://$HOME/workspace/conda-bld torchaudio
288288
- run:
289289
name: smoke test
@@ -325,7 +325,7 @@ jobs:
325325
conda env remove -n python${PYTHON_VERSION} || true
326326
conda create -yn python${PYTHON_VERSION} python=${PYTHON_VERSION}
327327
conda activate python${PYTHON_VERSION}
328-
conda install -v -y -c pytorch-${UPLOAD_CHANNEL} pytorch
328+
conda install -v -y -c pytorch-${UPLOAD_CHANNEL} pytorch cpuonly
329329
conda install -v -y $(ls ~/workspace/torchaudio*.tar.bz2)
330330
- run:
331331
name: smoke test

‎.circleci/config.yml.in

+2-2
Original file line numberDiff line numberDiff line change
@@ -283,7 +283,7 @@ jobs:
283283
command: |
284284
set -x
285285
source /usr/local/etc/profile.d/conda.sh && conda activate python${PYTHON_VERSION}
286-
conda install -v -y -c pytorch-${UPLOAD_CHANNEL} pytorch
286+
conda install -v -y -c pytorch-${UPLOAD_CHANNEL} pytorch cpuonly
287287
conda install -v -y -c file://$HOME/workspace/conda-bld torchaudio
288288
- run:
289289
name: smoke test
@@ -325,7 +325,7 @@ jobs:
325325
conda env remove -n python${PYTHON_VERSION} || true
326326
conda create -yn python${PYTHON_VERSION} python=${PYTHON_VERSION}
327327
conda activate python${PYTHON_VERSION}
328-
conda install -v -y -c pytorch-${UPLOAD_CHANNEL} pytorch
328+
conda install -v -y -c pytorch-${UPLOAD_CHANNEL} pytorch cpuonly
329329
conda install -v -y $(ls ~/workspace/torchaudio*.tar.bz2)
330330
- run:
331331
name: smoke test

0 commit comments

Comments
 (0)
Please sign in to comment.