From 2077d7888082dbd02fb68b3d6e4bde1decdc8e0c Mon Sep 17 00:00:00 2001
From: Glenn Jocher <glenn.jocher@ultralytics.com>
Date: Sat, 15 Aug 2020 09:53:22 -0700
Subject: [PATCH] update Dockerfile with libgl1-mesa-glx

---
 Dockerfile | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/Dockerfile b/Dockerfile
index f04a753..6193f55 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -5,6 +5,9 @@ FROM nvcr.io/nvidia/pytorch:20.03-py3
 COPY requirements.txt .
 RUN pip install -r requirements.txt gsutil
 
+# ImportError: libGL.so.1: cannot open shared object file: No such file or directory solution
+RUN apt update && apt install -y libgl1-mesa-glx
+
 # Create working directory
 RUN mkdir -p /usr/src/app
 WORKDIR /usr/src/app