python3.7+pytorch1.7.1+cuda10.2環境安裝torch_geometric


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

 


免責聲明!

本站轉載的文章為個人學習借鑒使用,本站對版權不負任何法律責任。如果侵犯了您的隱私權益,請聯系本站郵箱yoyou2525@163.com刪除。



 
粵ICP備18138465號   © 2018-2025 CODEPRJ.COM