Update Dockerfile

85zhanghao · Dec 12, 2022 · f9b106a · f9b106a
1 parent 2717112
commit f9b106a
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/Dockerfile b/Dockerfile
@@ -1,3 +1,6 @@
 FROM  pytorch/pytorch:1.10.0-cuda11.3-cudnn8-devel
-RUN pip install opencv-python
-RUN pip install albumentations
+RUN pip install opencv-python  opencv-contrib-python
+RUN pip install Cython matplotlib tqdm  numpy  scipy  PyYAML
+RUN pip install tensorboardX  tensorboard  scikit-learn  scikit-image
+RUN pip install seaborn  pillow   einops  lpips  natsort
+RUN pip install timm  openpyxl  kornia