FROM nvcr.io/nvidia/pytorch:20.12-py3 (#1620)

This commit is contained in:
Glenn Jocher 2020-12-22 17:48:40 -08:00 committed by GitHub
parent 1c39505d4e
commit 6ba36265fb
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -1,10 +1,13 @@
# Start FROM Nvidia PyTorch image https://ngc.nvidia.com/catalog/containers/nvidia:pytorch
FROM nvcr.io/nvidia/pytorch:20.10-py3
FROM nvcr.io/nvidia/pytorch:20.12-py3
# Install dependencies
# Install linux packages
RUN apt update && apt install -y screen libgl1-mesa-glx
# Install python dependencies
RUN pip install --upgrade pip
# COPY requirements.txt .
# RUN pip install -r requirements.txt
COPY requirements.txt .
RUN pip install -r requirements.txt
RUN pip install gsutil
# Create working directory