From a4f7767d3e45b793b050c394d47fa8a1d44cc574 Mon Sep 17 00:00:00 2001
From: 85zhanghao <94844844+85zhanghao@users.noreply.github.com>
Date: Sun, 30 Oct 2022 10:18:25 +0800
Subject: [PATCH] Update Dockerfile

---
 Dockerfile | 7 ++++++-
 1 file changed, 6 insertions(+), 1 deletion(-)

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