zjc
/
yolov5
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

FROM nvcr.io/nvidia/pytorch:20.10-py3 (#1553) 

* FROM pytorch/pytorch:latest

* FROM nvcr.io/nvidia/pytorch:20.10-py3
This commit is contained in:
Glenn Jocher 2020-11-29 17:47:51 +01:00 committed by GitHub
parent cff9263490
commit 68211f72c9
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
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.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