最近课题学习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
输出如下:
继续输入y,确认下载:
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环境是否安装成功
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"
安装结果如图所示:
8.安装 torch_geometric
(py36) D:\360downloads>pip install torch-geometric -i https://pypi.douban.com/simple
安装过程如下:
安装结果:成功安装pytorch_geometric
记录下来,希望也可以帮到坑里的其他小伙伴。