Pytorch无法运行在GPU上,提示显卡版本太低的解决方法
程序员文章站
2022-03-04 08:59:56
Pytorch无法运行在GPU,提示显卡版本太低的解决方法利用下列语句可以让pytorch选择运行在cpu或gpu上:DEVICE = torch.device("cuda" if torch.cuda.is_available() else "cpu") # 让torch判断是否使用GPU当提示无法运行在GPU:显卡版本太低。用以下方法解决:首先要检查配置CUDA-Cudnn-Pytorch,可以参考该文章:https://blog.csdn.net/qq_36653505/article/d...
Pytorch无法运行在GPU上,提示显卡版本太低的解决方法
利用下列语句可以让pytorch选择运行在cpu或gpu上:
DEVICE = torch.device("cuda" if torch.cuda.is_available() else "cpu")
# 让torch判断是否使用GPU
当提示无法运行在GPU:显卡版本太低。用以下方法解决:
1、检查Pytorch和Cuda的安装情况
用这两个命令可以分别检查pytorch和cuda安装是否成功,返回值如图所示则正常:
环境的配置可以参考该文章:https://blog.csdn.net/qq_36653505/article/details/83932941。我个人的步骤为:安装Anaconda,单独配置一个装有pytorch的python环境,安装PyCharm,将解释器设置为pytorch环境中的python。
2、升级显卡版本
不建议直接去官网下载,有时候速度慢,有时候会中途断连,当然也可以尝试,有时候比较顺利。
可以使用设备管理器的显示适配器-独显驱动升级来完成,快速稳定。
本文地址:https://blog.csdn.net/QB_Gan/article/details/107282295
上一篇: Flask中间件