Browse Source

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

5.0
Glenn Jocher 4 years ago
parent
commit
62e037419f
1 changed files with 4 additions and 6 deletions
  1. +4
    -6
      Dockerfile

+ 4
- 6
Dockerfile 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

Loading…
Cancel
Save