liav
/
docker_template
2
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

updated pytorch image

Browse Source
master
Vladimir Protsenko 2 weeks ago
parent 8053b7209b
commit b151bedd4a
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File

3
Dockerfile
Unescape Escape View File

@ -1,4 +1,5 @@
FROM pytorch/pytorch:2.2.0-cuda12.1-cudnn8-runtime FROM pytorch/pytorch:2.8.0-cuda12.9-cudnn9-runtime
# to compile with CUDA toolkit use pytorch/pytorch:2.8.0-cuda12.9-cudnn9-devel
ENV TZ=Europe/Samara ENV TZ=Europe/Samara
RUN ln -snf /usr/share/zoneinfo/$TZ /etc/localtime && echo $TZ > /etc/timezone RUN ln -snf /usr/share/zoneinfo/$TZ /etc/localtime && echo $TZ > /etc/timezone

Write Preview
Loading…
Cancel
Save