我命令行按照pytorch官网给的pip方式安装不行,于是进入下面的网址,troch、torchvision、torchaudio 的.whl文件下载地址下载.whl文件到本地,再安装。
1.查看NVIDIA驱动支持的CUDA版本
步骤:
桌面->右击鼠标->NVIDIA控制面板->帮助->系统信息->组件
可以将驱动更新到自己显卡支持的最高版本:
https://www.nvidia.cn/geforce/drivers/
至于如何查询显卡信息,步骤如下:
右击此电脑->管理->设备管理器->显示适配器
就可以看到
2.下载.whl文件并安装
进入下面的网址,troch、torchvision、torchaudio 的.whl文件下载地址下载.whl文件到本地
注:我下载的是上面三个文件,你可以在pytorch官网那个地方,把条件都选上,看它给的命令行里面需要下载几个文件就可,注意给的版本号
3.检查cuda
3.1 检测CUDA是否安装正确并能被Pytorch检测到
import torch
import torchvision
print(torch.cuda.is_available())
3.2 测试Pytorch能不能调用cuda加速
a = torch.Tensor(5,3)
a=a.cuda()
print(a)
我的3.1检测没问题,但3.2部分就会报错:
RuntimeError: CUDA error: no kernel image is available for execution on the device
应该是我的显卡的算力(3.5)不够,这也是知识目前这个原因猜测,具体原因还没弄明白,清楚的大佬可以教教我。
更新:开始是pytorch1.8,改为1.2就可以调用gpu。
最后在具体操作的过程在有什么问题,欢迎大家一起交流讨论。
在下编程小白,如果有什么错误欢迎大家批评指正!
邮箱:1916728303@qq.com