(小白自用保姆级经验教程)
本文介绍了使用GPU运行python的torch库的教程及遇到的一些问题
如何查看电脑的显卡配置命令
这一步是为了得到GPU算力从而安装相应版本的pytorch库,否则就会因为torch版本过高而渣电脑显卡太弱而遇到这个问题,这里表示我的显卡驱动程序版本是:457.51
查询了对应的显卡支持CUDA最高版本之后重新安装
查看教程如下:如何查看显卡支持的CUDA版本
然后就可以下载对应版本的torch了
pytorch下载方法:官方网站https://pytorch.org/
下载推荐使用anaconda Prompt窗口 成功率更高
全部安装好就可以
>>>import torch
>>>torch.cuda.is_available()
True
像这样就大功告成了