Dumpling Blog

Install Yolov8 With Anaconda

  • 929 字

使用軟體

anaconda | cuda-12-1 | pytorch

需要有nvidia顯卡並且支持cuda，才需安裝cuda

此教成演示版本為cuda12.1

如果使用其他版本的cuda，則需要到pytorch官網安裝對應版本的pytorch

創建環境

可以使用此命令查看cuda安裝是否成功及版本

nvcc --version

nvcc

創建環境

1
2
3
4
5
conda create -n yolov8 python=3.9
conda activate yolov8
conda install -c anaconda ipykernel
conda install cuda -c nvidia/label/cuda-12.1
conda install pytorch torchvision torchaudio pytorch-cuda=12.1 -c pytorch -c nvidia

可以使用此腳本看cuda是否在conda安裝成功

1
2
3
4
5
6
7
8
9
import torch

if torch.cuda.is_available():
    print("CUDA is available!")
    print("CUDA version:", torch.version.cuda)
    num_cuda_devices = torch.cuda.device_count()
    print("Number of CUDA devices available:", num_cuda_devices)
else:
    print("CUDA is not available.")

pycuda

安裝Yolov8

pip install ultralytics

查看Yolov8是否安裝成功

1
2
3
import ultralytics

ultralytics.checks()

Yolov8_check

如果有錯誤可以嘗試將charset-normalizer降版本

pip install --force-reinstall charset-normalizer==3.1.0

本作品採用 知識共用署名-相同方式共用 4.0 國際授權合約 進行許可
,