|
|
|
|
@ -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
|
|
|
|
|
RUN ln -snf /usr/share/zoneinfo/$TZ /etc/localtime && echo $TZ > /etc/timezone
|
|
|
|
|
|
|
|
|
|
|
