From d37accab46b55227d891eed6e9d99f4f0070e663 Mon Sep 17 00:00:00 2001 From: Tibor Reiss Date: Mon, 28 Oct 2024 21:53:09 +0100 Subject: [PATCH] Update docker files --- .circleci/docker/Dockerfile | 6 +++--- docker/dev/Dockerfile | 6 +++--- docker/release/Dockerfile | 6 +++--- 3 files changed, 9 insertions(+), 9 deletions(-) diff --git a/.circleci/docker/Dockerfile b/.circleci/docker/Dockerfile index 7212c32e9c..fb5f495608 100644 --- a/.circleci/docker/Dockerfile +++ b/.circleci/docker/Dockerfile @@ -1,6 +1,6 @@ -ARG PYTORCH="1.8.1" -ARG CUDA="10.2" -ARG CUDNN="7" +ARG PYTORCH="2.5.0" +ARG CUDA="12.4" +ARG CUDNN="9" FROM pytorch/pytorch:${PYTORCH}-cuda${CUDA}-cudnn${CUDNN}-devel diff --git a/docker/dev/Dockerfile b/docker/dev/Dockerfile index f63c284365..ba993e6e5c 100644 --- a/docker/dev/Dockerfile +++ b/docker/dev/Dockerfile @@ -1,6 +1,6 @@ -ARG PYTORCH="1.8.1" -ARG CUDA="10.2" -ARG CUDNN="7" +ARG PYTORCH="2.5.0" +ARG CUDA="12.4" +ARG CUDNN="9" FROM pytorch/pytorch:${PYTORCH}-cuda${CUDA}-cudnn${CUDNN}-devel diff --git a/docker/release/Dockerfile b/docker/release/Dockerfile index de099f4c2e..5b9520e02f 100644 --- a/docker/release/Dockerfile +++ b/docker/release/Dockerfile @@ -1,6 +1,6 @@ -ARG PYTORCH="1.8.1" -ARG CUDA="10.2" -ARG CUDNN="7" +ARG PYTORCH="2.5.0" +ARG CUDA="12.4" +ARG CUDNN="9" FROM pytorch/pytorch:${PYTORCH}-cuda${CUDA}-cudnn${CUDNN}-devel