Skip to content
Prev Previous commit
Next Next commit
attempt to fix cuda test
  • Loading branch information
Peter Romov committed Apr 5, 2022
commit b0bf58ba6f46b02d9c5e21d47466b521899443fe
10 changes: 10 additions & 0 deletions .circleci/config.yml
Original file line number Diff line number Diff line change
Expand Up @@ -40,6 +40,14 @@ commands:
name: "Install dependencies via pip"
command: ./scripts/install_via_pip.sh << parameters.args >>

torch_cuda_install:
description: "Install PyTorch"
steps:
- run:
name: "Install PyTorch"
command: |
pip install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu113

lint_flake8:
description: "Lint with flake8"
steps:
Expand Down Expand Up @@ -303,6 +311,7 @@ jobs:
steps:
- checkout
- py_3_7_setup
- torch_cuda_install
- pip_dev_install
- run_nvidia_smi
- mnist_integration_test:
Expand All @@ -323,6 +332,7 @@ jobs:
steps:
- checkout
- py_3_7_setup
- torch_cuda_install
- pip_dev_install
- run_nvidia_smi
- run:
Expand Down