Update Dockerfile

85zhanghao · Oct 30, 2022 · a4f7767 · a4f7767
1 parent e6b1fe8
commit a4f7767
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/Dockerfile b/Dockerfile
@@ -1 +1,6 @@
-FROM  pytorch/pytorch:1.8.0-cuda11.1-cudnn8-devel
+FROM  pytorch/pytorch:1.10.0-cuda11.3-cudnn8-devel
+RUN pip install opencv-python
+RUN apt-get update
+RUN apt --fix-broken install
+RUN apt-get install libgl1
+RUN pip install albumentations