admin管理员组文章数量:1562629
目录
- 准备
- 确认配置
- 首先，要确认自己电脑是64位的还是32位的：
- 其次，要确认自己电脑的GPU的CUDA版本：
- 最后，确认自己安装的python版本：
- 下载.whl包
- 安装.whl包
- 验证安装
- 参考资料
准备
确认配置
首先，要确认自己电脑是64位的还是32位的：
- 快捷键法：win+pause
- 我的电脑（此电脑） —— 右键 —— 属性
其次，要确认自己电脑的GPU的CUDA版本：
没有独立显卡的请在后面选择cpu版下载。
- 进入NVIDIA控制面板（以下方法仅供参考1，用其他方法能进入N卡控制面板也是可以的）
1）Win10搜索并进入控制面板；（快捷键win+S）
2）在控制面板右上角就搜索NVIDIA就能找到N卡的控制面板。
- 点击帮助(H)-系统信息(I)；
- 点击组件选项卡就可以看到CUDA版本了。
最后，确认自己安装的python版本：
使用win+R调出运行，并在运行中输入cmd，回车，在弹出的命令提示符窗口中输入以下语句：
python --version #注意是两个-
然后就会显示当前使用的python版本，还没有安装python的可以看这里。
下载.whl包
- 首先进入PyTorch的官网：pytorch
点击Get Started。 - 选择Windows，复制红线上的网址；
- 把复制好的地址粘贴到浏览器地址框，并回车访问，可以看到一整个网页的超链接；
- 使用Ctrl+F或F3调出查找框，并按如下规则找到需要的torch和torchvision（可以按第2步里官网推荐的版本）；
示例：（CUDA10.1，torch1.5.0，python3.7，Windows，64位）
5.下载找到的链接，这里建议使用IDMan下载器（请自行百度下载）
*基于IDMan下载器的下载方法：
1）右键链接地址，点击复制链接地址(E)；
2）进入IDMan，点击新建任务图标，将刚刚复制的链接粘贴进地址框，点击确定即可。
安装.whl包
找到刚刚下好的两个.whl文件所在的路径，使用win+R打开运行，输入cmd然后回车，来到命令控制符界面。
优先使用python -m pip install --upgrade pip命令更新一下自己的pip，待更新完成后使用
pip install 下载路径\***torch-***-cp**-cp**-win_amd64.whl
pip install 下载路径\***torchvision-***-cp**-cp**-win_amd64.whl
来安装本地whl包。
示例：（CUDA10.1，torch1.5.0，torchvision0.6.0，python3.7，Windows，64位，下载到E盘下）
pip install E:\torch-1.5.0+cu101-cp37-cp37m-win_amd64.whl
pip install E:\torchvision-0.6.0+cu101-cp37-cp37m-win_amd64.whl
之前直接使用pip再加上清华源下载，发现torch只有低版本的，可能是镜像没同步的原因吧。心想不管它直接pip install torch的时候，又出现了如下错误：
ModuleNotFoundError: No module named ‘tools.nnwrap’
最后上S.O.搜了一下2才有了上面这个方案。
验证安装
使用win+R打开运行，输入cmd然后回车，来到命令控制符界面。
输入pip list3找找有没有如下两个项目和对应的版本号，有即为安装成功。
参考资料
Windows系统查看CUDA版本号——潇洒坤 ↩︎
ModuleNotFoundError: No module named ‘tools.nnwrap’ (windows)——CDJB, Harish Vutukuri ↩︎
Pip怎么查看已安装的包？教你查看pip已装模块——BinnLZeng ↩︎
版权声明：本文标题：Win10使用pip安装PyTorch的方法简述 内容由热心网友自发贡献，该文观点仅代表作者本人， 转载请联系作者并注明出处：https://m.elefans.com/dianzi/1727485262a1117126.html， 本站仅提供信息存储空间服务，不拥有所有权，不承担相关法律责任。如发现本站有涉嫌抄袭侵权/违法违规的内容，一经查实，本站将立刻删除。
发表评论