ICode9

精准搜索请尝试: 精确搜索
首页 > 其他分享> 文章详细

pytorch1.5+torchvision0.6.0安装教程(win10)

2021-07-17 18:01:59  阅读:2138  来源: 互联网

标签:whl win torch win10 pytorch1.5 torchvision0.6 cp37 cuda10.1 下载


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

下载之前查看自己cuda版本,命令ncvv -V即可查看

作者使用的cuda10.1,下表为cuda与torch对应表,cuda10.1支持的torch版本还是比较多的

然后开始正式下载:

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

注意:cu101代表cuda10.1,cp37代表python版本,win_amd64代表是windows系统下载的

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

下载完之后,在下载的目录内使用下列命令即可:

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

标签:whl,win,torch,win10,pytorch1.5,torchvision0.6,cp37,cuda10.1,下载
来源: https://blog.csdn.net/qq_40639828/article/details/118860047

本站声明: 1. iCode9 技术分享网(下文简称本站)提供的所有内容,仅供技术学习、探讨和分享;
2. 关于本站的所有留言、评论、转载及引用,纯属内容发起人的个人观点,与本站观点和立场无关;
3. 关于本站的所有言论和文字,纯属内容发起人的个人观点,与本站观点和立场无关;
4. 本站文章均是网友提供,不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属;如您发现该文章侵犯了您的权益,可联系我们第一时间进行删除;
5. 本站为非盈利性的个人网站,所有内容不会用来进行牟利,也不会利用任何形式的广告来间接获益,纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。

专注分享技术,共同学习,共同进步。侵权联系[81616952@qq.com]

Copyright (C)ICode9.com, All Rights Reserved.

ICode9版权所有