一、安装ｃｕｄａ和ｃｕｄｎｎ

1、安装cuda之前先打开英伟达控制面板查看自己的显卡信息

2、“帮助”—>“系统信息”—>“组件”，然后看第三行的信息“Nvidia CUDA 12.4”

3、进入英伟达官网“CUDA Toolkit Archive | NVIDIA Developer”，我的是12.4

下载完成后nvcc -V检查一下是否安装正确

nvcc -V

查看是否配置环境变量

二、下载CUDNN

cuDNN Archive | NVIDIA Developer

下载好了过后把解压出来放到CUDAv12.4安装目录下

然后配置环境变量

D:\daima\CUDAv12.4\bin

D:\daima\CUDAv12.4\extras\CUPTI\lib64

三、安装pytorch（带GPU）

PyTorch

1、

2、

3、找到这个，我的是12.4，安装12.1的也行

4、安装ultralytics

pip install ultralytics

5、因为ultralytics自带是cpu的我们需要卸载cpu版本的pytorch

pip uninstall torch

6、安装第三步复制的内容

conda install pytorch==2.2.2 torchvision==0.17.2 torchaudio==2.2.2 pytorch-cuda=12.1 -c pytorch -c nvidia

7、可以测试一下了（如果不行看最下面）

import ultralytics
ultralytics.checks()

import torch
print(torch.__version__)
print(torch.cuda.is_available())
print(torch.cuda.device_count())

OK

安装后报错：ImportError: cannot import name 'COMMON_SAFE_ASCII_CHARACTERS' from 'charset_normalizer.constant' (D:\daima\anaconda3\yolo\lib\site-packages\charset_normalizer\constant.py)

pip install chardet

YOLO5模型训练过程_mac训练yolov5-CSDN博客

YOLO5 6.1版本报错解决_ranges = np.round(np.linspace(8, 4, 10), 0).astype-CSDN博客