yolov5遇到的问题-在anaconda 虚拟环境中安装pytorch，cuda

问题一：TypeError: vars() argument must have dict attribute

解决办法：

问题二：AssertionError: CUDA unavailable, invalid device 0 requested

cuda不可用报错，现实没有有效的驱动可使用

解决方案

1、测试cuda是否配置正确
1）在终端输入

python3

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

运行后输出的是False，说明cuda有问题。
继续找问题：有可能是pytorch的版本是cpu版本不是cuda版本的！

方法：查看torch的版本：

python
import torch
print(torch.version)
1.13.1+cpu

接下来安装cuda版本的torch—这里我是在线安装

1：查看cuda版本：nvidia-smi【注意这不是查看anaconda中的cuda版本】

nvidia-smi

查看anaconda环境中cuda的版本号：
nvcc --version:

3.输入如下指令查看是否安装CUDA

nvcc -V

若安装完成显示下图

3. 查看cudnn版本

import torch
print(torch.backends.cudnn.version())

下图显示:None,表示未安装成功。

问题：>>> import torch，torch.cuda.is_available()==False

cmd中检查cuda及cudnn是否成功安装及其版本

1. 在cmd中检查cuda是否安装成功：nvcc -V
   
   原文链接：https://blog.csdn.net/qq_40968179/article/detai