yolov5项目cuda错误解决

CUDA报错解决

# 报错详情
AssertionError: CUDA unavailable, invalid device 0 requested

查看cuda版本

先看一下电脑是否支持GPU，打开任务管理器就能查看（ctrl+shift+esc）

# cmd命令
nvcc --version
# 如果上面命令不是内部或外部命令，也不是可运行的程序，就输入下面的命令
NVIDIA-smi
# 如果没有cuda的话，可以官网直接下载（下载有点慢，多等待一下）
全部版本：https://developer.nvidia.com/cuda-toolkit-archive（选择一个想要的版本点进去）
这里示例11.7.0：https://developer.nvidia.com/cuda-11-7-0-download-archive

torch三件套安装

pytorch官网：https://pytorch.org/get-started/previous-versions/

找到指定的版本安装（最好是安装到虚拟环境中，Python安装虚拟环境：https://www.cnblogs.com/hkwJsxl/p/17211497.html）

因为是国外的网站，所以会慢一些，也可以多次下载失败，可以多尝试几次，或换一个时间段进行下载

Upsample报错解决

# 报错详情（我的项目是解决完上面的cuda报错后就出现了这个报错）
AttributeError: 'Upsample' object has no attribute 'recompute_scale_factor'
# 解决办法
点进去源码，把forward函数代码换成下面这一句
return F.interpolate(input, self.size, self.scale_factor, self.mode, self.align_corners)