diff --git a/Dockerfile b/Dockerfile
index 242af10..96bad94 100644
--- 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