Pytorch和torch_geometric安装（Windows）

一、引言和tips

记录一些经常用到的技巧和命令，可以快速部署环境，防止浪费生命！

（1）pip安装python 库（清华源）

1）安装numpy（替换库）为例：

pip install numpy -i https://pypi.tuna.tsinghua.edu.cn/simple

2）安装依赖文件：

pip install -r requirements.txt -i https://pypi.tuna.tsinghua.edu.cn/simple

3）阿里源：

pip install tb-nightly -i http://mirrors.aliyun.com/pypi/simple/ --trusted-host mirrors.aliyun.com

（2）查看电脑GPU显卡的版本

Nvidia-smi

可以直接看到自己的电脑cuda版本，方便后续安装pytorch。

二、Anaconda虚拟环境命令（可选）

打开anaconda prompt命令行（有报错，权限不够，用管理员身份进入）

（1）修改anaconda镜像源为国内（清华源）

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/

（2）虚拟环境命令

1）创建虚拟环境

conda create --name myenv python=3.10

2）进入虚拟环境

conda activate myenv

3）退出虚拟环境

conda deactivate

三、Pytorch安装

（1）打开官网地址

Start Locally | PyTorch

如下图所示，选择对应的系统（这里是Windows版本），用pip安装（conda比较慢）：

直接将命令

pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu121

复制到anaconda prompt里（要进入你的虚拟环境里），等待安装完成。

（2）查看pytorch版本

在虚拟环境中的anaconda prompt里输入以下命令，并回车

python

进入python后，可以知道自己的python版本，输入命令：

import torch
print(torch.__version__)                # 查看pytorch安装的版本号
print(torch.cuda.is_available())        # 查看cuda是否可用。True为可用，即是gpu版本pytorch
print(torch.cuda.get_device_name(0))    # 返回GPU型号
print(torch.cuda.device_count())        # 返回可以用的cuda（GPU）数量，0代表一个
print(torch.version.cuda)               # 查看cuda的版本

四、Torch_geometric安装

（1）直接pip安装

在国内，如果直接输入以下命令，正常是安装失败的：

pip install torch_geometric

解决方法也很简单，就是离线安装。

（2）离线whl安装

1）先进入pyg的官网whl地址：data.pyg.org/whl/

2）根据我们自己的电脑系统版本、python版本、CPU（GPU）版本、torch版本下载对应的whl：

首先，根据torch版本、cpu/cuda版本选择：

举个例子：这里的torch-2.3.0+cu121是说torch版本是2.3.0，cuda版本是12.1。

然后点击进去，

我们要下载四个whl（版本越新越好）：torch_cluster、torch_scatter、torch_sparse和torch_spline_conv。

然后根据你的电脑版本、python版本、cpu/cuda版本选择：

举个例子：这里的torch_scatter-2.1.2+pt23cu121-cp310-cp310-win_amd64.whl是说torch_scatter版本是2.1.2，cuda版本是12.1，python版本是3.10系列的，电脑系统版本是Windows x64。

下载到你看得到的文件目录中。

3）安装whl文件和pyg库

回到anaconda prompt（虚拟环境中的）进入到下载的目录中，输入：

pip install xxx.whl -i https://pypi.tuna.tsinghua.edu.cn/simple

注意：不要修改文件名字，要原原本本。

最后再输入：

pip install torch_geometric -i https://pypi.tuna.tsinghua.edu.cn/simple

在pip list中可以看到安装后的pyg库。