HuQiong
/
yolov8_test
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

FROM pytorch/pytorch:2.2.2-cuda12.1-cudnn8-runtime (#9452)

Browse Source
main
Glenn Jocher 6 months ago committed by GitHub
parent cd172e9d12
commit 30484d5925
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 1 additions and 1 deletions
Whitespace
Show all changes Ignore whitespace when comparing lines Ignore changes in amount of whitespace Ignore changes in whitespace at EOL
Split View
Diff Options
Show Stats Download Patch File Download Diff File
  1. 2
      docker/Dockerfile

2
docker/Dockerfile
Unescape View File

@ -3,7 +3,7 @@
# Image is CUDA-optimized for YOLOv8 single/multi-GPU training and inference
# Start FROM PyTorch image https://hub.docker.com/r/pytorch/pytorch or nvcr.io/nvidia/pytorch:23.03-py3
FROM pytorch/pytorch:2.2.0-cuda12.1-cudnn8-runtime
FROM pytorch/pytorch:2.2.2-cuda12.1-cudnn8-runtime
RUN pip install --no-cache nvidia-tensorrt --index-url https://pypi.ngc.nvidia.com
# Downloads to user config dir

Write Preview
Loading…
Cancel
Save