1. Pytorch 和 PyG
Pytorch和PyG是深度學習的python環境,后者用於圖處理。其主要特點如下:
①需要與cuda(NVIDIA)配合,必須先裝載cuda。
②cuda,pytorch,pytorch_geometric三者之間版本依賴精密,必須屬於同一套版本。
其安裝順序即 cuda > pytorch > PyG
③其不能通過pycharm進行package安裝,必須先安裝在環境中,再由pycharm關聯使用該環境。
④推薦使用anoconda創立環境並下載如上兩包。
2. 配置步驟
2.1 cuda的安裝
CUDA(Compute Unified Device Architecture),是顯卡廠商NVIDIA推出的運算平台。 CUDA是一種由NVIDIA推出的通用並行計算架構,該架構使GPU能夠解決復雜的計算問題。
詳見如下博客:
https://blog.csdn.net/NSJim/article/details/115408138
2.2 anoconda 環境激活
使用Anoconda prompt(windows內搜索)進入anoconda下的命令行,據此創建/激活環境,並進行導包等操作。

詳見鏈接:
https://blog.csdn.net/SARACH_WONG/article/details/89328307
2.3 pytorch的安裝
注:pytorch對cuda的版本是向下兼容的,即可以使用更低等級的cuda,如11.1的cuda,可以安裝對應10.2版本cuda的pytorch。
- 進入Pytorch官網(https://pytorch.org/),選擇對應的版本號(可向下兼容),這里我們選擇穩定版本1.10.2和cuda10.2
- 將圖示中Run this Command所提供的代碼在Anoconda激活的環境中運行(即
activate phm2012phm2012為環境名),安裝pytorch。

2.4 Pytorch_Geometri的安裝
同理,基於上述基礎,在Anoconda中進行PyG的導入:
- 進入PyG官網(https://pytorch-geometric.readthedocs.io/en/latest/notes/installation.html)
- 選擇版本號,在Anoconda激活的環境中安裝。

2.5 Pycharm中找到該環境
在pycharm中找到配置好的anoconda環境即可。
