Update Dockerfile

85zhanghao · Dec 12, 2023 · b20665d · b20665d
1 parent 4ee3508
commit b20665d
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/Dockerfile b/Dockerfile
@@ -1,2 +1,3 @@
-FROM 85zhanghao/docker:v3.2.0
-RUN pip install timm  einops
+FROM pytorch/pytorch:1.12.1-cuda11.3-cudnn8-devel
+RUN pip install albumentations==1.0.0 einops==0.4.1 fast_pytorch_kmeans==0.1.6 glob2==0.7  gpustat==0.6.0 h5py==3.6.0 matplotlib==3.3.4  numpy==1.22.4
+RUN pip install tqdm==4.64.1  timm==0.9.12    termcolor==2.4.0   tensorboard==2.12.2