带GPU的Pytorch安装【看看避坑啊!】

之前写过一篇如何安装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
posted @ 2022-11-25 17:47  芋圆院长  阅读(166)  评论(0编辑  收藏  举报