Update Dockerfile FROM nvcr.io/nvidia/pytorch:20.08-py3

This commit is contained in:
Glenn Jocher committed 2020-08-26 00:44:34 -07:00
1 parent 31d6c389f2
commit 62e037419f
1 file changed
+4 -6
+4 -6
View File
@@ -1,12 +1,10 @@
# Start FROM Nvidia PyTorch image https://ngc.nvidia.com/catalog/containers/nvidia:pytorch
FROM nvcr.io/nvidia/pytorch:20.03-py3
FROM nvcr.io/nvidia/pytorch:20.08-py3
# Install dependencies
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
# COPY requirements.txt .
# RUN pip install -r requirements.txt
RUN pip install gsutil
# Create working directory
RUN mkdir -p /usr/src/app