commit 94c444f4d7739d80d91f7c90943cb0b350807761 Author: mingzailao Date: Mon Mar 13 19:56:38 2023 +0800 sd 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 + + + +