1、官網上查看自己的環境對應的torch_geometric版本
(1)github網址:https://github.com/rusty1s/pytorch_geometric
(2)打開網站之后,能在網頁上看到 “Installation”,我用到的版本如下圖:
2、安裝torch_geometric的命令
我的電腦配置是cuda10.2,對應的是cu102(如果不是這個環境,可以更改下面命令中加粗的內容),在命令行中運行如下代碼即可:
pip install torch-scatter -f https://pytorch-geometric.com/whl/torch-1.7.0+cu102.html
pip install torch-sparse -f https://pytorch-geometric.com/whl/torch-1.7.0+cu102.html
pip install torch-cluster -f https://pytorch-geometric.com/whl/torch-1.7.0+cu102.html
pip install torch-spline-conv -f https://pytorch-geometric.com/whl/torch-1.7.0+cu102.html
pip install torch-geometric
3、是否需要其他依賴包?
在運行上述命令的過程中,我發現系統會自動下載一些包,通過查看原項目,可以看到有一個setup.py文件,打開后可以看到下圖,numpy, tqdm等都是本項目的依賴包。如果之前沒有下載過這些包也不用擔心,直接運行第二步驟的代碼,系統會自動安裝需要的包。
4、安裝出現問題,未安裝上torch-cluster和torch-scatter
可以打開網址https://pytorch-geometric.com/whl/torch-1.7.0.html(如果是其他版本的torch,使用該網址查看對應版本https://pytorch-geometric.com/whl/),找自己需要的版本下載之后放在“script"文件夾下,也可以自建文件夾,我的虛擬環境的配置是python3.7.9,pytorch版本是1.7.1,查找到對應的兼容標簽(Compatible tags)是”cp37-cp37m-win_amd64“,因此下載了下圖所示的版本。
總結安裝步驟:
- 下載合適的torch-cluster版本,並放在“script"文件夾下
- cd到該“script"文件夾
- 輸入命令pip install torch_cluster-1.5.8+cpu-cp37-cp37m-win_amd64.whl,並回車
- 顯示Successfully...... 安裝成功!!!
注意:查找到對應的兼容標簽的方法:參考https://blog.csdn.net/happywlg123/article/details/107281936
我們可以通過下面這個經過測試有效的命令,來查看當前Python版本匹配那些文件。
有效命令如下(直接在命令行中輸入,不要在Python解釋器輸入哦):
pip debug --verbose
結果很長,但是我們在這里需要的是”Compatible tags"后面的內容,如下,紅圈中就是我們需要的版本。
一定要注意版本的匹配,如果使用cu102版本的torch-geometric,就要使用cu102版本的torch-cluster和torch-scatter