Windows下安装pytorch教程(下载.whl的方式)

阅读: 评论:0

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

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

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

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

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

pytorch与torchvision下载链接: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

本文发布于:2025-03-21 13:29:00,感谢您对本站的认可!

本文链接:https://www.4u4v.net/it/1742534947582740.html

版权声明:本站内容均来自互联网,仅供演示用,请勿用于商业和其他非法用途。如果侵犯了您的权益请与我们联系,我们将在24小时内删除。

标签:方式   教程   Windows   pytorch   whl
留言与评论(共有 0 条评论)
   
验证码:

Copyright ©2019-2022 Comsenz Inc.Powered by ©

网站地图1 网站地图2 网站地图3 网站地图4 网站地图5 网站地图6 网站地图7 网站地图8 网站地图9 网站地图10 网站地图11 网站地图12 网站地图13 网站地图14 网站地图15 网站地图16 网站地图17 网站地图18 网站地图19 网站地图20 网站地图21 网站地图22/a> 网站地图23