对于标题的报错,首先确认自己电脑是否有GPU,驱动时是否正常

然后根据GPU的型号安装cuda、cuDNN、和GPU版本的torch

安装cuda
安装CUDNN
设置环境变量
NVIDIA官网安装cuda对应的GPU版本的torch
具体安装可参考:
https://blog.csdn.net/qq_62102643/article/details/130041247

更多推荐