之前写过一篇如何安装Pytorch的博客,但主要是CPU版本的,注意的事项还比较少。现在专门再记录一遍安装GPU版本的。

CPU版本参考:Pytorch的安装

前言

一定要记住,先看自己电脑有没有英伟达的显卡。没有,跳转至CPU。

有显卡时,查看显卡算力,不到3.0的,跳转至CPU。

接下来要做的第一件事是确认自己要安装的PYtorch版本和Python版本(这两个一定要对上啊)
以我为例,要安装的是PyTorch 1.4 和 Python 3.6。

常规确认步骤(参考链接1即可)

【这些常规步骤都是按照电脑配置本身来的。如果你要复现别人的东西按照别人的配置来,那只能倒着来…】
1.根据NVIDIA控制面板,查看CUDA所支持的最高版本(系统信息 –> 组件)
pytorch 1.4.0 对应的CUDA环境有9.2 和10.1【我选择10.1】
经过确认,显卡版本可以与10.1对应。

  1. 跳转到该链接下载对应的CUDA https://developer.nvidia.com/cuda-toolkit-archive

  2. 跳转到对应链接下载cuDAA (全看参考链接1就好啦!真的不踩雷!)配好这些东西以后。

直接在Anaconda Prompt中创建环境加上对应的Python 版本就好啦!
Pytorch版本看我之前整理的链接就好了。

本来以为这次又要很久很久,但结果还好欸~

希望大家都能顺利安装。

参考链接

  1. 【带 GPU 的 Pytorch 安装】https://blog.csdn.net/qq_38140292/article/details/114157146

原文地址:http://www.cnblogs.com/yuyuanliu/p/16925912.html

1. 本站所有资源来源于用户上传和网络,如有侵权请邮件联系站长! 2. 分享目的仅供大家学习和交流,请务用于商业用途! 3. 如果你也有好源码或者教程,可以到用户中心发布,分享有积分奖励和额外收入! 4. 本站提供的源码、模板、插件等等其他资源,都不包含技术服务请大家谅解! 5. 如有链接无法下载、失效或广告,请联系管理员处理! 6. 本站资源售价只是赞助,收取费用仅维持本站的日常运营所需! 7. 如遇到加密压缩包,默认解压密码为"gltf",如遇到无法解压的请联系管理员! 8. 因为资源和程序源码均为可复制品,所以不支持任何理由的退款兑现,请斟酌后支付下载 声明:如果标题没有注明"已测试"或者"测试可用"等字样的资源源码均未经过站长测试.特别注意没有标注的源码不保证任何可用性