本文主要是介绍Pytorch安装小坑(Windows+cu111),希望对大家解决编程问题提供一定的参考价值,需要的开发者们随着小编来一起学习吧!
从Pytorch官网根据cuda和Anaconda创建的Python环境,发现需要执行以下命令:
pip install torch==1.10.1+cu111 torchvision==0.11.2+cu111 torchaudio==0.10.1 -f
但在安装完torch后,会发现,没有对应的torchvision,然后进行不下去了
后查看官方的包目录,有点逆天了,没有对应Win的包
最后只能安装这个:
pip install torch==1.9.1+cu111 torchvision==0.10.1+cu111 torchaudio==0.9.1 -f https://download.pytorch.org/whl/torch_stable.html
验证程序:
import torch
x = torch.rand(5, 3)
print(x)
这篇关于Pytorch安装小坑(Windows+cu111)的文章就介绍到这儿,希望我们推荐的文章对编程师们有所帮助!