From 94c444f4d7739d80d91f7c90943cb0b350807761 Mon Sep 17 00:00:00 2001 From: mingzailao Date: Mon, 13 Mar 2023 19:56:38 +0800 Subject: [PATCH] sd --- Dockerfile | 7 +++++++ 1 file changed, 7 insertions(+) create mode 100644 Dockerfile diff --git a/Dockerfile b/Dockerfile new file mode 100644 index 0000000..70b9548 --- /dev/null +++ b/Dockerfile @@ -0,0 +1,7 @@ +FROM mingzailao/cuda:11.3.1-cudnn8-devel-ubuntu20.04 + +RUN pip3 install torch==1.12.1+cu113 torchvision==0.13.1+cu113 torchaudio==0.12.1 --extra-index-url https://download.pytorch.org/whl/cu113 + + + +