背景

今天运行基于pytorch的yolov5时报了warning

warnings.warn(‘User provided device_type of ‘cuda’, but CUDA is not available. Disabling’)

检验cuda和pytorch是否兼容

import torch

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

我的输出结果如下，可以看出我的pytorch和cuda版本不兼容

解决方法

1. 打开pytorch官网 pytorch官网地址
2. 按照自己电脑的配置选择，我是cuda10.2，windows电脑，安装的anaconda按下图选择
3. 打开Anaconda Prompt，激活对应的环境
   
4. 将pytorch官网给出的命令复制上，pip3改为pip（根据自己的pip版本）
5. 安装成功，显示True，torch为cuda加速版本