Update Dockerfile

85zhanghao · Oct 10, 2023 · 91e124f · 91e124f
1 parent a447d14
commit 91e124f
Showing 1 changed file with 2 additions and 6 deletions.
diff --git a/Dockerfile b/Dockerfile
@@ -1,8 +1,4 @@
-FROM pytorch/pytorch:1.11.0-cuda11.3-cudnn8-runtime
+FROM 85zhanghao/docker:v3.2.0
 SHELL ["/bin/bash", "-c"]
 
-RUN apt-get update
-RUN apt-get upgrade -y
-RUN apt-get install -y apt-utils wget unzip git
-RUN pip install --upgrade pip
-RUN pip install tqdm yacs>=0.1.8 timm>=0.5.4 numpy==1.21.5
+RUN pip install timm  einops