deep_learning/ultralytics
1
0
Fork 0
mirror of https://github.com/ultralytics/ultralytics.git synced 2025-09-15 15:48:41 +08:00
Code Issues Actions 2 Packages Projects Releases Wiki Activity

Fix latest-nvidia-cuda Docker tag (#19835)

Browse Source 
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
This commit is contained in:
Glenn Jocher 2025-03-23 23:50:29 +01:00 committed by GitHub
parent e5921b472f
commit 6c80ba6e95
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
.github/workflows/docker.yml vendored
View File

@ -66,7 +66,7 @@ jobs:
platforms: "linux/amd64"
runs_on: "ubuntu-latest"
- dockerfile: "Dockerfile-nvidia-cuda"
tags: "latest"
tags: "latest-nvidia-cuda"
platforms: "linux/amd64"
runs_on: "ubuntu-latest"
- dockerfile: "Dockerfile-cpu"

3
docker/Dockerfile-nvidia-cuda
View File

@ -3,7 +3,8 @@
# Builds ultralytics/ultralytics:latest-nvidia-cuda image on DockerHub https://hub.docker.com/r/ultralytics/ultralytics
# Image is CUDA-optimized for YOLO11 single/multi-GPU training and inference
# Start FROM PyTorch image https://hub.docker.com/r/pytorch/pytorch or nvcr.io/nvidia/pytorch:25.02-py3 or nvidia/cuda:12.8.1-cudnn-runtime-ubuntu24.04
# Start FROM PyTorch image https://hub.docker.com/r/pytorch/pytorch or nvcr.io/nvidia/pytorch:25.02-py3 or nvidia/cuda:12.8.1-cudnn-runtime-ubuntu22.04
# Warning: ubuntu24.04 does not work (TFLite bug)
FROM nvidia/cuda:12.8.1-cudnn-runtime-ubuntu22.04
# Set environment variables