之前写过一篇如何安装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对应。
-
跳转到该链接下载对应的CUDA https://developer.nvidia.com/cuda-toolkit-archive
-
跳转到对应链接下载cuDAA (全看参考链接1就好啦!真的不踩雷!)配好这些东西以后。
直接在Anaconda Prompt中创建环境加上对应的Python 版本就好啦!
Pytorch版本看我之前整理的链接就好了。
本来以为这次又要很久很久,但结果还好欸~
希望大家都能顺利安装。
参考链接
- 【带 GPU 的 Pytorch 安装】https://blog.csdn.net/qq_38140292/article/details/114157146