From 7a0e70717a7492b64e682c49e75e24e0bbcea0be Mon Sep 17 00:00:00 2001
From: mingzailao <mingzailao@126.com>
Date: Tue, 28 Mar 2023 15:52:06 +0800
Subject: [PATCH] sda

---
 Dockerfile | 6 ++++--
 1 file changed, 4 insertions(+), 2 deletions(-)

diff --git a/Dockerfile b/Dockerfile
index 65cff00..61f259d 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -1,6 +1,8 @@
-FROM mingzailao/cuda:11.3.1-cudnn8-devel-ubuntu20.04
+FROM mingzailao/cuda:10.2-cudnn8-devel-ubuntu18.04
 
-RUN pip3 install torch==1.10.1+cu113 torchvision==0.11.2+cu113 torchaudio==0.10.0 --extra-index-url https://download.pytorch.org/whl/cu113
+
+
+RUN pip3 install torch==1.9.1+cu102  --extra-index-url https://download.pytorch.org/whl/cu102