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
pip3 install torch==1.10.0+cu102 torchvision==0.11.1+cu102 torchaudio===0.10.0+cu102 -f https://download.pytorch.org/whl/cu102/torch_stable.html
(成功的选择)
通过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也能运行,作为菜鸟的我也不知道为什么,反正成功了
(官网)
(自己运行)