Use pytorch with cuda12.4 (#2168)

jupyter · Nov 4, 2024 · 2c525af · 2c525af
1 parent c2d9d1f
commit 2c525af
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/images/pytorch-notebook/cuda12/Dockerfile b/images/pytorch-notebook/cuda12/Dockerfile
@@ -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/cu121' \
+RUN pip install --no-cache-dir --extra-index-url=https://pypi.nvidia.com --index-url 'https://download.pytorch.org/whl/cu124' \
     'torch' \
     'torchaudio' \
     'torchvision' && \