From 4785451644a043ecf51043d3568c6f313abee1cc Mon Sep 17 00:00:00 2001 From: ldwang Date: Tue, 7 Feb 2023 17:37:36 +0800 Subject: [PATCH] Update Dockerfile --- Dockerfile | 1 + 1 file changed, 1 insertion(+) diff --git a/Dockerfile b/Dockerfile index c29f12e5..31c9d211 100644 --- a/Dockerfile +++ b/Dockerfile @@ -1,4 +1,5 @@ #Change to your base image, such as pytorch1.11+py38 +#https://docs.nvidia.com/deeplearning/frameworks/pytorch-release-notes/rel_21-02.html#rel_21-02 FROM nvcr.io/nvidia/pytorch:21.06-py3 #You can set available pypi sources RUN /bin/bash -c "pip config set global.index-url https://pypi.tuna.tsinghua.edu.cn/simple"