hazza/docker-stacks
1
0
Fork 0
mirror of https://github.com/jupyter/docker-stacks.git synced 2025-10-17 15:02:57 +00:00
Code Issues Packages Projects Releases Wiki Activity

Upgrade CUDA version used in PyTorch cuda12-variant to 12.6 (#2291)

Browse Source
This commit is contained in:
Ayaz Salikhov
2025-04-25 01:39:58 +01:00
committed by GitHub
parent 1dbce17442
commit 99bd3f8a24
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
images/pytorch-notebook/cuda12/Dockerfile
View File

@@ -13,7 +13,7 @@ SHELL ["/bin/bash", "-o", "pipefail", "-c"]
# Install PyTorch with pip (https://pytorch.org/get-started/locally/)
# hadolint ignore=DL3013
RUN pip install --no-cache-dir --extra-index-url=https://pypi.nvidia.com --index-url 'https://download.pytorch.org/whl/cu124' \
RUN pip install --no-cache-dir --extra-index-url=https://pypi.nvidia.com --index-url 'https://download.pytorch.org/whl/cu126' \
'torch' \
'torchaudio' \
'torchvision' && \