Skip to content

Commit 45fdd2e

Browse files
committed
chore(Dockerfile): default to PyTorch 1.11.0
1 parent d0a6dfe commit 45fdd2e

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

docker/zettasets/Dockerfile

+2-2
Original file line numberDiff line numberDiff line change
@@ -8,9 +8,9 @@ RUN apt-get --allow-releaseinfo-change update && \
88
git clone https://${GIT_ACCESS_TOKEN}@github.com/ZettaAI/zettasets.git && \
99
rm -rf /var/lib/apt/lists/*
1010

11-
# FROM pytorch/pytorch:1.11.0-cuda11.3-cudnn8-runtime
11+
FROM pytorch/pytorch:1.11.0-cuda11.3-cudnn8-runtime
1212
# FROM pytorch/pytorch:1.2-cuda10.0-cudnn7-runtime
13-
FROM pytorch/pytorch:2.0.1-cuda11.7-cudnn8-runtime
13+
# FROM pytorch/pytorch:2.0.1-cuda11.7-cudnn8-runtime
1414
# FROM pytorch/pytorch:2.1.0-cuda12.1-cudnn8-runtime
1515
ARG DEBIAN_FRONTEND=noninteractive
1616

0 commit comments

Comments
 (0)