FROM nvcr.io/nvidia/pytorch:21.10-py3 (#6673)
Reordered installation may help reduce resource usage in autobuildpull/6677/head
parent
25456284b4
commit
18b5081166
|
@ -1,7 +1,7 @@
|
||||||
# YOLOv5 🚀 by Ultralytics, GPL-3.0 license
|
# YOLOv5 🚀 by Ultralytics, GPL-3.0 license
|
||||||
|
|
||||||
# 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:21.10-py3
|
FROM nvcr.io/nvidia/pytorch:22.01-py3
|
||||||
|
|
||||||
# Install linux packages
|
# Install linux packages
|
||||||
RUN apt update && apt install -y zip htop screen libgl1-mesa-glx
|
RUN apt update && apt install -y zip htop screen libgl1-mesa-glx
|
||||||
|
|
Loading…
Reference in New Issue