From 384b5769531bd7f22a48c5ddcbb5e53b5f1b97c0 Mon Sep 17 00:00:00 2001 From: Fangjun Kuang Date: Sun, 22 Oct 2023 20:40:19 +0800 Subject: [PATCH] small fixes --- docker/torch2.1.0-cuda11.8.dockerfile | 2 +- docker/torch2.1.0-cuda12.1.dockerfile | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/docker/torch2.1.0-cuda11.8.dockerfile b/docker/torch2.1.0-cuda11.8.dockerfile index ed9a7464f..f130ab0de 100644 --- a/docker/torch2.1.0-cuda11.8.dockerfile +++ b/docker/torch2.1.0-cuda11.8.dockerfile @@ -1,4 +1,4 @@ -FROM pytorch/pytorch:2.1.0-cuda11.8-cudnn8-devel +FROM pytorch/pytorch:2.1.0-cuda11.8-cudnn8-runtime ENV LC_ALL C.UTF-8 diff --git a/docker/torch2.1.0-cuda12.1.dockerfile b/docker/torch2.1.0-cuda12.1.dockerfile index 4d8dd86c5..9f38ebb99 100644 --- a/docker/torch2.1.0-cuda12.1.dockerfile +++ b/docker/torch2.1.0-cuda12.1.dockerfile @@ -1,4 +1,4 @@ -FROM pytorch/pytorch:2.1.0-cuda12.1-cudnn8-devel +FROM pytorch/pytorch:2.1.0-cuda12.1-cudnn8-runtime ENV LC_ALL C.UTF-8