栏目分类:
子分类:
返回
名师互学网用户登录
快速导航关闭
当前搜索
当前分类
子分类
实用工具
热门搜索
名师互学网 > IT > 软件开发 > 后端开发 > Python

torch.cuda.is

Python 更新时间: 发布时间: IT归档 最新发布 模块sitemap 名妆网 法律咨询 聚返吧 英语巴士网 伯小乐 网商动力

torch.cuda.is

pytorch官网:

Start Locally | PyTorch

一定要去官网复制pip命令行,别人的不一定和你版本相同

下载cuda前一定先看一下putorch支持的版本,不然下了也是白下

首先,最开始的疑惑来自于安装完成后

 通过网上各种教程还未解决,直到最后使用官网命令

(这是失败的选择)

 运行结果出现了

ERROR: Could not find a version that satisfies the requirement torchaudio===0.10.0+cu113 ERROR: No matching distribution

的(没截图)错误提示

到官网后发现有个模块根本没有资源,后来降低cuda版本后,成功的通过Pip安装成功,并且torch.cuda.is_available()返回了True

(成功的选择)

通过nvidia-smi查看版本及驱动

 

(但是我的cuda版本是11.3,我也不是到为什么选择10.2成功了) 

寻找三个模块

Pip命令行可以看见安装了三个模块(菜鸟的说法,大佬绕道)

分别是torch==1.10.0+cu102、torchvision==0.11.1+cu102、torchaudio===0.10.0+cu102三个,而我的问题就是最后一个找不到,并且我手动寻找也没找到

 

这便是三个模块的下载地址, 

 复制https://download.pytorch.org/whl/cu102/进入网页

 

可以看见根本没有torchaudio===0.10.0+cu102 的win版本,所以再怎么安装都是无用的,无奈只能选择降低cuda版本进行安装

本来准备重新安装cuda的,降低版本到10.2,没想到我的11.3也能运行,作为菜鸟的我也不知道为什么,反正成功了

 

(官网)

(自己运行)

 

 

 

 

 

转载请注明:文章转载自 www.mshxw.com
本文地址:https://www.mshxw.com/it/348290.html
我们一直用心在做
关于我们 文章归档 网站地图 联系我们

版权所有 (c)2021-2022 MSHXW.COM

ICP备案号:晋ICP备2021003244-6号