admin 管理员组

文章数量: 1086019

本文以pytorch1.5为例，通过下载whl文件的方式，如需安装其他版本，步骤一样

下载pytorch1.5的同时，我们附带下载torchvision0.6.0

①首先查看自己cuda版本，命令ncvv -V即可查看

②作者使用的cuda10.1，查看支持的pytorch版本，cuda10.1支持的torch版本还是比较多的

③然后我选择了pytorch1.5开始正式下载：

pytorch与torchvision下载链接：https://download.pytorch/whl/torch_stable.html

进入下载页面之后，ctrl+f寻找你要下载的pytorch版本，（注意：cu101代表cuda10.1，cp37代表python版本，win_amd64代表是windows系统下载的），找到之后点击下载

④我对应的就是红色箭头指的。同理找torchvision，这个是专门用来处理图像的库，下载比较快

⑤下载完pytorch和torchvision的.whl文件之后，在下载的目录内使用下列命令即可：

pip install torchvision-0.6.0+cu101-cp37-cp37m-win_amd64.whl
pip install torch-1.5.0+cu101-cp37-cp37m-win_amd64.whl

本文标签： 方式 教程 Windows pytorch whl

版权声明：本文标题：Windows下安装pytorch教程(下载.whl的方式) 内容由网友自发贡献，该文观点仅代表作者本人， 转载请联系作者并注明出处：http://www.roclinux.cn/b/1742534948a2405240.html， 本站仅提供信息存储空间服务，不拥有所有权，不承担相关法律责任。如发现本站有涉嫌抄袭侵权/违法违规的内容，一经查实，本站将立刻删除。