使用GPU预训练模型的时候报错 No supported gpu backend found!
然后根据GPU的型号安装cuda、cuDNN、和GPU版本的torch。对于标题的报错,首先确认自己电脑是否有GPU,驱动时是否正常。NVIDIA官网安装cuda对应的GPU版本的torch。
·
对于标题的报错,首先确认自己电脑是否有GPU,驱动时是否正常
然后根据GPU的型号安装cuda、cuDNN、和GPU版本的torch
安装cuda
安装CUDNN
设置环境变量
NVIDIA官网安装cuda对应的GPU版本的torch
具体安装可参考:
https://blog.csdn.net/qq_62102643/article/details/130041247
更多推荐
所有评论(0)