最近課題學(xué)習(xí)pytorch Geometric，需要安裝環(huán)境，記錄一下。
一、版本介紹
操作系統(tǒng)：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
此過(guò)程已經(jīng)默認(rèn)cuda10.1和cudnn10.1已經(jīng)配置好了。
二、操作過(guò)程
1.打開(kāi)Anaconda Prompt，創(chuàng)建python3.6虛擬環(huán)境，輸入下面的命令

conda create -n py36 python=3.6.5

輸出如下：

圖1.1

繼續(xù)輸入y，確認(rèn)下載：

圖1.2

圖1.3 py3.6虛擬環(huán)境創(chuàng)建成功

2.激活虛擬環(huán)境
輸入下面命令，進(jìn)入py36環(huán)境（base)->(py36)

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

3.安裝numpy 1.18.1(3,4步為基礎(chǔ)）

(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的安裝過(guò)程及結(jié)果如下所示：

以上安裝成功

5.下載版本匹配的相關(guān)whl文件
首先需要在py36環(huán)境下，將目錄切換到whl文件所在路徑，我的所有whl文件是在D盤 360downloads文件夾下。

切換文件路徑

(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環(huán)境是否安裝成功

pytorch安裝成功的效果

打開(kāi)python,可以引入pytorch

7.PYG相關(guān)包的安裝
將目錄切換到這四個(gè)文件所在的路徑下，輸入下列命令：

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"

安裝結(jié)果如圖所示：

image.png

image.png

8.安裝 torch_geometric

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

安裝過(guò)程如下：

image.png

image.png

image.png

image.png

安裝結(jié)果：成功安裝pytorch_geometric

成功安裝提示

打開(kāi)python，引入pytorch_geometric

記錄下來(lái)，希望也可以幫到坑里的其他小伙伴。