|
|
@ -1,5 +1,5 @@
|
|
|
|
# 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.03-py3
|
|
|
|
FROM nvcr.io/nvidia/pytorch:20.06-py3
|
|
|
|
RUN pip install -U gsutil
|
|
|
|
RUN pip install -U gsutil
|
|
|
|
|
|
|
|
|
|
|
|
# Create working directory
|
|
|
|
# Create working directory
|
|
|
|