computervisioneng/train-yolov8-custom-dataset-step-by-step-guide

How to use Yolov8 with cuda?

wolf002 opened this issue · 1 comments

wolf002 commented

How to intasll Yolov8 with support cuda? Please descibe.

Gormonboza commented

Before you run pip install ultralytics:

  1. install CUDA toolkit (for example CUDA 11.8) and CUDNN
  2. install pyTorch [https://pytorch.org/get-started/locally/]