Skip to content
Prev Previous commit
Next Next commit
attempt $6
  • Loading branch information
Peter Romov committed Apr 8, 2022
commit 4ebadd005f4982738906f0a3f2783b5e2f302c35
6 changes: 3 additions & 3 deletions .circleci/config.yml
Original file line number Diff line number Diff line change
Expand Up @@ -46,7 +46,7 @@ commands:
- run:
name: "Install PyTorch"
command: |
pip install torch torchvision torchaudio
pip install torch==1.10.2 torchvision torchaudio

lint_flake8:
description: "Lint with flake8"
Expand Down Expand Up @@ -307,7 +307,7 @@ jobs:
integrationtest_py37_torch_release_cuda:
machine:
resource_class: gpu.nvidia.small.multi
image: ubuntu-2004-cuda-11.2:202103-01
image: ubuntu-1604-cuda-10.2:202012-01
steps:
- checkout
- py_3_7_setup
Expand All @@ -328,7 +328,7 @@ jobs:
unittest_multi_gpu:
machine:
resource_class: gpu.nvidia.medium.multi
image: ubuntu-2004-cuda-11.2:202103-01
image: ubuntu-1604-cuda-10.2:202012-01
steps:
- checkout
- py_3_7_setup
Expand Down