Glenn Jocher
GitHub
3 years ago
1 changed files with 1 additions and 1 deletions
+ 1
- 1
Dockerfile
View File
| @@ -1,5 +1,5 @@ | |||
| # Start FROM Nvidia PyTorch image https://ngc.nvidia.com/catalog/containers/nvidia:pytorch | |||
| FROM nvcr.io/nvidia/pytorch:20.12-py3 | |||
| FROM nvcr.io/nvidia/pytorch:21.02-py3 | |||
| # Install linux packages | |||
| RUN apt update && apt install -y zip htop screen libgl1-mesa-glx | |||
Loading…