Skip to content

Commit 365fedd

Browse files
authored
Update and rename cu12.1-py310-torch2.1.1-jupyterlab-vscode.Dockerfile to cu12.1-py310-torch2.2.0-jupyterlab-vscode.Dockerfile
1 parent 279f70c commit 365fedd

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

Diff for: image/cu12.1-py310-torch2.1.1-jupyterlab-vscode.Dockerfile renamed to image/cu12.1-py310-torch2.2.0-jupyterlab-vscode.Dockerfile

+2-2
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,4 @@
1-
FROM nvidia/cuda:12.1.1-cudnn8-devel-ubuntu20.04
1+
FROM nvidia/cuda:12.3.2-cudnn9-devel-ubuntu22.04
22

33
RUN apt update && \
44
apt install -y \
@@ -26,7 +26,7 @@ RUN rm -r /temp && \
2626
ln -s /usr/local/bin/pip3 /usr/local/bin/pip
2727

2828
# install pytorch
29-
RUN pip install torch==2.1.0 torchvision==0.16.0 torchaudio==2.1.0 --index-url https://download.pytorch.org/whl/cu121 && \
29+
RUN pip install torch==2.2.0 torchvision==0.17.0 torchaudio==2.2.0 --index-url https://download.pytorch.org/whl/cu121 && \
3030
rm -r /root/.cache/pip
3131

3232
#

0 commit comments

Comments
 (0)