From 68211f72c99915a15855f7b99bf5d93f5631330f Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Sun, 29 Nov 2020 17:47:51 +0100 Subject: [PATCH] FROM nvcr.io/nvidia/pytorch:20.10-py3 (#1553) * FROM pytorch/pytorch:latest * FROM nvcr.io/nvidia/pytorch:20.10-py3 --- Dockerfile | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/Dockerfile b/Dockerfile index 005d1af..7a3e0f8 100644 --- a/Dockerfile +++ b/Dockerfile @@ -1,5 +1,5 @@ # Start FROM Nvidia PyTorch image https://ngc.nvidia.com/catalog/containers/nvidia:pytorch -FROM nvcr.io/nvidia/pytorch:20.11-py3 +FROM nvcr.io/nvidia/pytorch:20.10-py3 # Install dependencies RUN pip install --upgrade pip @@ -28,7 +28,7 @@ COPY . /usr/src/app # for v in {300..303}; do t=ultralytics/coco:v$v && sudo docker build -t $t . && sudo docker push $t; done # Pull and Run -# t=ultralytics/yolov5:latest && sudo docker pull $t && sudo docker run -it --ipc=host $t +# t=ultralytics/yolov5:latest && sudo docker pull $t && sudo docker run -it --ipc=host --gpus all $t # Pull and Run with local directory access # t=ultralytics/yolov5:latest && sudo docker pull $t && sudo docker run -it --ipc=host --gpus all -v "$(pwd)"/coco:/usr/src/coco $t