apoorva/greenhouse
2
1
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

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

Browse Source
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
1 changed files with 7 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
Dockerfile
View File

@ -1,10 +1,13 @@
# Start FROM Nvidia PyTorch image https://ngc.nvidia.com/catalog/containers/nvidia:pytorch # 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 RUN pip install --upgrade pip
# COPY requirements.txt . COPY requirements.txt .
# RUN pip install -r requirements.txt RUN pip install -r requirements.txt
RUN pip install gsutil RUN pip install gsutil
# Create working directory # Create working directory