今天在用conda配置的yolov8环境跑代码时候遇到了这个报错,cuda版本是11.4,torch版本是2.0.1
2.x的torch至少需要cuda11.7及其以上吧?cuda是向下兼容,不是向上,也就是你环境里面的cuda版本比torch要求的高才行(适用于cuda11,cuda12没试过,cuda10.x不行)。
另外,nvidia-smi命令显示的是你的硬件驱动最高支持的cuda,不表示你环境里面安装的cuda版本,查看具体cuda版本使用nvcc -V命令,这个才是你环境里面目前链接的cuda版本
D:\Users\Downloads>python
Python 3.7.0 (default, Jun 28 2018, 08:04:48) [MSC v.1912 64 bit (AMD64)] :: Anaconda, Inc. on win32
Type "help", "copyright", "credits" or "license" for more information.
>>> import torch
>>> print(torch.cuda.is_available())
>>> yes