Skip to content

Commit 0f8e47b

Browse files
authored
Fix failing GPU ci (#2759)
* Update config.yml * Updates in .github/workflows/gpu-tests.yml
1 parent 117529e commit 0f8e47b

File tree

2 files changed

+5
-5
lines changed

2 files changed

+5
-5
lines changed

.circleci/config.yml

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -4,11 +4,11 @@ parameters:
44
pytorch_stable_image:
55
type: string
66
# https://hub.docker.com/r/pytorch/pytorch/tags
7-
default: "pytorch/pytorch:1.12.1-cuda11.3-cudnn8-runtime"
7+
default: "pytorch/pytorch:1.13.0-cuda11.6-cudnn8-runtime"
88
pytorch_stable_image_devel:
99
type: string
1010
# https://hub.docker.com/r/pytorch/pytorch/tags
11-
default: "pytorch/pytorch:1.12.1-cuda11.3-cudnn8-devel"
11+
default: "pytorch/pytorch:1.13.0-cuda11.6-cudnn8-devel"
1212
workingdir:
1313
type: string
1414
default: "/tmp/ignite"

.github/workflows/gpu-tests.yml

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -49,16 +49,16 @@ jobs:
4949
${{ steps.pip-cache.outputs.dir }}
5050
key: ${{ steps.get-date.outputs.date }}-pytorch-${{ runner.os }}-3.8-pytorch-${{ hashFiles('requirements-dev.txt') }}
5151
restore-keys: |
52-
${{ steps.get-date.outputs.date }}-pytorch-${{ runner.os }}--3.8-pytorch-
52+
${{ steps.get-date.outputs.date }}-pytorch-${{ runner.os }}-3.8-pytorch-
5353
5454
- run: pip install pip wheel setuptools -Uqq
5555

5656
- name: Install PyTorch
5757
# https://pytorch.org/get-started/locally/
5858
run: |
59-
pip install torch torchvision -f https://download.pytorch.org/whl/cu116/torch_stable.html
59+
pip install --upgrade torch torchvision -f https://download.pytorch.org/whl/cu116/torch_stable.html
6060
nvidia-smi
61-
python -c "import torch; print('CUDA is available: ', torch.cuda.is_available())"
61+
python -c "import torch; print(torch.__version__, ', CUDA is available: ', torch.cuda.is_available())"
6262
6363
- name: Install dependencies
6464
run: |

0 commit comments

Comments
 (0)