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:
parent
cff9263490
commit
68211f72c9
|
|
@ -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.11-py3
|
FROM nvcr.io/nvidia/pytorch:20.10-py3
|
||||||
|
|
||||||
# Install dependencies
|
# Install dependencies
|
||||||
RUN pip install --upgrade pip
|
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
|
# for v in {300..303}; do t=ultralytics/coco:v$v && sudo docker build -t $t . && sudo docker push $t; done
|
||||||
|
|
||||||
# Pull and Run
|
# 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
|
# 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
|
# t=ultralytics/yolov5:latest && sudo docker pull $t && sudo docker run -it --ipc=host --gpus all -v "$(pwd)"/coco:/usr/src/coco $t
|
||||||
|
|
|
||||||
Loading…
Reference in New Issue