From 89a2fb3453d1553cd290c58849f377d17464de8a Mon Sep 17 00:00:00 2001 From: Daniel Dale Date: Wed, 20 Dec 2023 12:50:24 -0800 Subject: [PATCH] update cuda dockerfile to use pytorch test channel --- .github/workflows/ci_test-full.yml | 2 +- dockers/base-cuda/Dockerfile | 4 ++-- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/.github/workflows/ci_test-full.yml b/.github/workflows/ci_test-full.yml index 60ea33b..08b1e3a 100644 --- a/.github/workflows/ci_test-full.yml +++ b/.github/workflows/ci_test-full.yml @@ -42,7 +42,7 @@ jobs: matrix: os: [ubuntu-20.04, windows-2022, macOS-11] python-version: ["3.8", "3.11"] # minimum, maximum - # tests will include oldest and newest tested patch versions of pytorch-lightning + # tests will include oldest and newest tested patch versions of pytorch requires: ["oldest", "latest"] release: ["latest"] exclude: # note that occasionally, the oldest PyTorch version does not support the latest python version diff --git a/dockers/base-cuda/Dockerfile b/dockers/base-cuda/Dockerfile index 6a4d2e1..8d96547 100644 --- a/dockers/base-cuda/Dockerfile +++ b/dockers/base-cuda/Dockerfile @@ -91,9 +91,9 @@ RUN \ # ... pytorch patch version # pip install torch==1.11.1+cu113 torchvision==0.11.3+cu113 -f https://download.pytorch.org/whl/cu113/torch_stable.html; \ # ... pytorch nightly dev version - pip install --pre torch==2.2.0.dev20231021 torchvision==0.17.0.dev20231021 -f https://download.pytorch.org/whl/nightly/cu121/torch_nightly.html; \ + # pip install --pre torch==2.2.0.dev20231021 torchvision==0.17.0.dev20231021 -f https://download.pytorch.org/whl/nightly/cu121/torch_nightly.html; \ # ... test channel - # pip install --pre torch torchvision -f https://download.pytorch.org/whl/test/cu121/torch_test.html; \ + pip install --pre torch torchvision -f https://download.pytorch.org/whl/test/cu121/torch_test.html; \ fi && \ # Install all requirements pip install -r requirements/devel.txt --no-cache-dir && \