From 1ed921bff794954a7b882475861b4d7a9317bf9a Mon Sep 17 00:00:00 2001 From: hoshi-hiyouga Date: Sat, 2 Nov 2024 20:20:26 +0800 Subject: [PATCH] Update Dockerfile Former-commit-id: 89a1c1eb6d717b20107c06a645652b87fba388e8 --- docker/docker-cuda/Dockerfile | 2 ++ 1 file changed, 2 insertions(+) diff --git a/docker/docker-cuda/Dockerfile b/docker/docker-cuda/Dockerfile index f4c12462..34503290 100644 --- a/docker/docker-cuda/Dockerfile +++ b/docker/docker-cuda/Dockerfile @@ -1,3 +1,5 @@ +# Default use the NVIDIA official image with PyTorch 2.3.0 +# https://docs.nvidia.com/deeplearning/frameworks/pytorch-release-notes/index.html ARG BASE_IMAGE=nvcr.io/nvidia/pytorch:24.02-py3 FROM ${BASE_IMAGE}