.whl文件本地化安装pytorch

我命令行按照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

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值