win10+Anaconda 3+cuda10.1+cudnn10.1+py36+torch1.4.0+torchvision0.5.0+Pytorch_Geometric 详细安装过程

最近课题学习pytorch Geometric，需要安装环境，记录一下。
一、版本介绍
操作系统：win10
Anaconda 3
cuda_10.1.168_425.25_win10
cudnn-10.1-windows10-x64-v7.6.5.32
python 3.6
torch-1.4.0-cp36-cp36m-win_amd64.whl
torchvision 0.5.0
PYG需要用到的包：
torch_cluster-1.5.2+cu101-cp36-cp36m-win_amd64.whl
torch_scatter-2.0.4+cu101-cp36-cp36m-win_amd64.whl
torch_sparse-0.6.0+cu101-cp36-cp36m-win_amd64.whl
torch_spline_conv-1.2.0+cu101-cp36-cp36m-win_amd64.whl
此过程已经默认cuda10.1和cudnn10.1已经配置好了。
二、操作过程
1.打开Anaconda Prompt，创建python3.6虚拟环境，输入下面的命令

conda create -n py36 python=3.6.5

输出如下：

图1.1

继续输入y，确认下载：

图1.2

图1.3 py3.6虚拟环境创建成功

2.激活虚拟环境
输入下面命令，进入py36环境（base)->(py36)

(base) C:\Windows\system32>activate py36

3.安装numpy 1.18.1(3,4步为基础）

(py36) C:\Windows\system32>pip install numpy==1.18.1 -i https://pypi.douban.com/simple

4.安装pillow 7.0.0

(py36) C:\Windows\system32>pip install pillow==7.0.0 -i https://pypi.douban.com/simple

步骤3，步骤4的安装过程及结果如下所示：

以上安装成功

5.下载版本匹配的相关whl文件
首先需要在py36环境下，将目录切换到whl文件所在路径，我的所有whl文件是在D盘 360downloads文件夹下。

切换文件路径

6.通过pip install 分别安装 torch 和 torchvision（pytorch两个基础）

(py36) D:\360downloads>pip install torch-1.4.0-cp36-cp36m-win_amd64.whl

(py36) D:\360downloads>pip install torchvision==0.5.0 -i https://pypi.douban.com/simple

输入命令 nvcc -V,检查pytorch环境是否安装成功

pytorch安装成功的效果

打开python,可以引入pytorch

7.PYG相关包的安装
将目录切换到这四个文件所在的路径下，输入下列命令：

pip install "torch_scatter-2.0.4+cu101-cp36-cp36m-win_amd64.whl"

pip install "torch_sparse-0.6.0+cu101-cp36-cp36m-win_amd64.whl"

pip install "torch_spline_conv-1.2.0+cu101-cp36-cp36m-win_amd64.whl"

安装结果如图所示：

image.png

image.png

8.安装 torch_geometric

(py36) D:\360downloads>pip install torch-geometric -i https://pypi.douban.com/simple

安装过程如下：

image.png

image.png

image.png

image.png

安装结果：成功安装pytorch_geometric

成功安装提示

打开python，引入pytorch_geometric

记录下来，希望也可以帮到坑里的其他小伙伴。