Skip to content
Prev Previous commit
Next Next commit
attempt #3
  • Loading branch information
Peter Romov committed Apr 5, 2022
commit a28f71c2cf85ed0e3500c13f7114256908e21ab1
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 torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu113

lint_flake8:
description: "Lint with flake8"
Expand Down Expand Up @@ -308,7 +308,7 @@ jobs:
machine:
resource_class: gpu.nvidia.small.multi
#image: ubuntu-2004-cuda-11.4:202110-01
image: ubuntu-1604-cuda-10.2:202012-01
image: nvidia/cuda:11.3.1-base-ubuntu20.04
steps:
- checkout
- py_3_7_setup
Expand All @@ -330,7 +330,7 @@ jobs:
machine:
resource_class: gpu.nvidia.medium.multi
#image: ubuntu-2004-cuda-11.4:202110-01
image: ubuntu-1604-cuda-10.2:202012-01
image: nvidia/cuda:11.3.1-base-ubuntu20.04
steps:
- checkout
- py_3_7_setup
Expand Down