首先这个问题是因为cuda与显卡不适配,或者cuda与pytorch版本不适配问题;
首先查看是否是cuda与显卡不适配的问题,建议直接看Z大佬的文章:CUDA版本与显卡驱动匹配_显卡驱动对应的cuda-优快云博客
如果以上问题解决之后,仍有问题,那就是cuda与pytorch版本不适配问题,可以更新pytorch
进入pytorch官网:PyTorch
选择跟自己电脑以及cuda版本相适应的版本,在anaconda里进行安装
重新运行,问题解决。
首先这个问题是因为cuda与显卡不适配,或者cuda与pytorch版本不适配问题;
首先查看是否是cuda与显卡不适配的问题,建议直接看Z大佬的文章:CUDA版本与显卡驱动匹配_显卡驱动对应的cuda-优快云博客
如果以上问题解决之后,仍有问题,那就是cuda与pytorch版本不适配问题,可以更新pytorch
进入pytorch官网:PyTorch
选择跟自己电脑以及cuda版本相适应的版本,在anaconda里进行安装
重新运行,问题解决。