Update and rename cu12.1-py310-torch2.1.1-jupyterlab-vscode.Dockerfil…

…e to cu12.1-py310-torch2.2.0-jupyterlab-vscode.Dockerfile
NCHU-NLP-Lab · Mar 6, 2024 · 365fedd · 365fedd
1 parent 279f70c
commit 365fedd
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/...0-torch2.1.1-jupyterlab-vscode.Dockerfile → ...0-torch2.2.0-jupyterlab-vscode.Dockerfile b/...0-torch2.1.1-jupyterlab-vscode.Dockerfile → ...0-torch2.2.0-jupyterlab-vscode.Dockerfile
@@ -1,4 +1,4 @@
-FROM nvidia/cuda:12.1.1-cudnn8-devel-ubuntu20.04
+FROM nvidia/cuda:12.3.2-cudnn9-devel-ubuntu22.04
 
 RUN apt update && \
     apt install -y \
@@ -26,7 +26,7 @@ RUN rm -r /temp && \
     ln -s /usr/local/bin/pip3 /usr/local/bin/pip
 
 # install pytorch
-RUN pip install torch==2.1.0 torchvision==0.16.0 torchaudio==2.1.0 --index-url https://download.pytorch.org/whl/cu121 && \
+RUN pip install torch==2.2.0 torchvision==0.17.0 torchaudio==2.2.0 --index-url https://download.pytorch.org/whl/cu121 && \
     rm -r /root/.cache/pip
 
 #