FROM nvcr.io/nvidia/pytorch:20.10-py3 (#1553)
Browse files* FROM pytorch/pytorch:latest
* FROM nvcr.io/nvidia/pytorch:20.10-py3
- Dockerfile +2 -2
Dockerfile
CHANGED
|
@@ -1,5 +1,5 @@
|
|
| 1 |
# Start FROM Nvidia PyTorch image https://ngc.nvidia.com/catalog/containers/nvidia:pytorch
|
| 2 |
-
FROM nvcr.io/nvidia/pytorch:20.
|
| 3 |
|
| 4 |
# Install dependencies
|
| 5 |
RUN pip install --upgrade pip
|
|
@@ -28,7 +28,7 @@ COPY . /usr/src/app
|
|
| 28 |
# for v in {300..303}; do t=ultralytics/coco:v$v && sudo docker build -t $t . && sudo docker push $t; done
|
| 29 |
|
| 30 |
# Pull and Run
|
| 31 |
-
# t=ultralytics/yolov5:latest && sudo docker pull $t && sudo docker run -it --ipc=host $t
|
| 32 |
|
| 33 |
# Pull and Run with local directory access
|
| 34 |
# t=ultralytics/yolov5:latest && sudo docker pull $t && sudo docker run -it --ipc=host --gpus all -v "$(pwd)"/coco:/usr/src/coco $t
|
|
|
|
| 1 |
# Start FROM Nvidia PyTorch image https://ngc.nvidia.com/catalog/containers/nvidia:pytorch
|
| 2 |
+
FROM nvcr.io/nvidia/pytorch:20.10-py3
|
| 3 |
|
| 4 |
# Install dependencies
|
| 5 |
RUN pip install --upgrade pip
|
|
|
|
| 28 |
# for v in {300..303}; do t=ultralytics/coco:v$v && sudo docker build -t $t . && sudo docker push $t; done
|
| 29 |
|
| 30 |
# Pull and Run
|
| 31 |
+
# t=ultralytics/yolov5:latest && sudo docker pull $t && sudo docker run -it --ipc=host --gpus all $t
|
| 32 |
|
| 33 |
# Pull and Run with local directory access
|
| 34 |
# t=ultralytics/yolov5:latest && sudo docker pull $t && sudo docker run -it --ipc=host --gpus all -v "$(pwd)"/coco:/usr/src/coco $t
|