Windows 10下安装GPU版pytorch和torch-geometric

---

前言

2022.05.12
前两天在家查找了一晚上安装攻略后没有安装成功，今天重新整理电脑后安装成功，故在此记录一下安装过程。
显卡：3060
显卡驱动版本：512.59

---

一、安装显卡驱动

1、首先先把电脑里面和驱动相关程序全部卸载，卸载驱动可以使用DDU（我不清楚到底是原因导致安装一直失败，所以就把电脑清理了一遍，老版的DDU是可以用的，如果不行的话就更新最新版本）：

链接：https://pan.baidu.com/s/1yNgccFQ-Ge_90jwtgx84tA
提取码：feyw

卸载完成后打开电脑应用于功能查看是否有残留显卡驱动，将带有NVIDIA的应用全部卸载后重启

重启后进入NVIDIA官网重新下载驱动，下载最新版的驱动
（我下载的驱动是22年4月26日的512.59）

二、安装Anaconda

Anaconda安装比较简单，网上有很多教程，直接到官网下载安装即可

https://www.anaconda.com/

安装完成后的python是最新版本，可以进行合适版本进行配置：

1、安装合适的python版本

（我比较喜欢python 3.8，所以安装的是python 3.8）
在开始的最近添加找到Anaconda Prompt(Anaconda)并打开，依次输入：

conda create --name py3.8 python=3.8
conda activate py3.8

即可。最近添加栏里会多出Anaconda Prompt（py3.8）

2、配置清华镜像源

打开Anaconda Prompt（py3.8）并输入：

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/r
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/msys2
conda config --set show_channel_urls yes

添加完成后查看镜像源可输入：

conda config --show

添加完成后输入：

conda install anaconda

安装常用的包（numpy、pandas等包）

三、安装Pytorch

进入官网 pytorch.org 主页，选择合适的版本（我的版本如图）

我看别人帖子上说30系的显卡要用CUDA 11+，所以选择了CUDA11.3，选择完毕后使用打开Anaconda Prompt（py3.8）输入图中最后一行命令：

conda install pytorch torchvision torchaudio cudatoolkit=11.3 -c pytorch

（有时下载较慢，晚上的时候下载失败了，今天白天又成功了，神奇）
安装完成后打开python输入

import torch
print(torch.cuda.is_available())

输出 True 即为成功

此时深度学习模型就可以在GPU中训练了，如需研究图神经网络，可以继续安装torch-geometric

四、安装torch-geometric

在此之前，安装CPU版的torch-geometric老是出错，后来阅读一篇帖子说

使用Anaconda安装，使用Anaconda Prompt运行对应版本cmd
首先安装Pytorch==1.8.0，不要安装1.8.1和最新版本
然后运行代码：
conda install pyg -c pyg -c conda-forge

安装成功了，如果需要安装CPU版本可以尝试一下。
这次安装直接查询 pytorch-geometric 官网
https://pytorchgeometric.readthedocs.io/en/latest/notes/installation.html
选择对应的pytorch版本（我安装的是pytorch1.11，CUDA11.3）

同样在Anaconda Prompt（py3.8）中输入:

conda install pyg -c pyg

即可安装，安装完成后打开python输入

import torch_geometric

不报错即为成功，就可以开始构建自己的模型啦。

总结

自己喜欢折腾，但总是喜欢忘事，在此记录一下，也为有需要的同学提供一个思路。文笔不好，请谅解