Skip to content

Commit

Permalink
update cuda dockerfile to use pytorch test channel
Browse files Browse the repository at this point in the history
  • Loading branch information
speediedan committed Dec 20, 2023
1 parent 5889c84 commit 89a2fb3
Show file tree
Hide file tree
Showing 2 changed files with 3 additions and 3 deletions.
2 changes: 1 addition & 1 deletion .github/workflows/ci_test-full.yml
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand Down
4 changes: 2 additions & 2 deletions dockers/base-cuda/Dockerfile
Original file line number Diff line number Diff line change
Expand Up @@ -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 && \
Expand Down

0 comments on commit 89a2fb3

Please sign in to comment.