Update Dockerfile

zhanghang1989 · Feb 11, 2023 · 6dcb4cc · 6dcb4cc
1 parent 6bfa68b
commit 6dcb4cc
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/Dockerfile b/Dockerfile
@@ -8,8 +8,8 @@ WORKDIR /workspace
 RUN apt-get update && DEBIAN_FRONTEND=noninteractive apt-get install -y python3-tk python-pip git tmux htop tree
 
 RUN python -m pip install --upgrade pip
-RUN python -m pip install torch==1.4.0
-RUN python -m pip install torchvision==0.5.0
+#RUN python -m pip install torch==1.4.0
+#RUN python -m pip install torchvision==0.5.0
 RUN python -m pip install pycocotools==2.0.0
 
 #RUN chmod a+rwx -R /opt/conda/