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

GNU Affero General Public License v3.0
257 stars 112 forks source link

How to use Yolov8 with cuda? #1

Open wolf002 opened 1 year ago

wolf002 commented 1 year ago

How to intasll Yolov8 with support cuda? Please descibe.

Gormonboza commented 7 months ago

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/]