PyG文件之一 安裝

2021-10-08 22:05:47 字數 1714 閱讀 6712

2.cpu安裝(推薦新手)

3.檢查安裝結果

pyg是torch-geometric的簡稱,類似於numpy,pyg已經成為了圖神經網路的一大殺器。安裝pyg需要預先安裝torch,如需使用gpu,則還需要安裝cuda。這兩者的安裝很簡單,不再贅述。初學者建議只安裝cpu版本,會簡單一些。最好再安裝乙個anaconda來管理虛擬環境。

我的演示環境:win10+conda+notebook,顯示卡是mx150

linux下則預設你已經會安裝了。

相關url:

cuda官方**

torch官方**

pyg官方文件

#$表示視窗命令

$ python -c "import torch; print(torch.__version__)"

>>> 1.6.0

$ python -c "import torch; print(torch.version.cuda)"

>>> 10.2

pip install torch-scatter==latest+cu102 -f 

pip install torch-sparse==latest+cu102 -f

pip install torch-cluster==latest+cu102 -f

pip install torch-spline-conv==latest+cu102 -f

pip install torch-geometric

如果你的版本是cuda92+torch1.5.0,則你應該使用如下安裝語句:

$ pip install torch-scatter==latest+cu92 -f
其它的以此推類。如果遇到找不到對應包的錯誤,則建議換乙個torch或者cuda的版本。

我一開始是cuda92,但是安裝後出現了問題。推薦版本:cuda:10.2 torch:1.5/1.6

若使用cpu安裝,則需要安裝cpu版本的pytorch,為了不破壞原有環境,建議新建乙個虛擬環境。

$ conda create -n pyg_cpu

#啟用虛擬環境

$ conda activate pyg_cpu

$ conda install pytorch torchvision cpuonly -c pytorch

$ python -c "import torch; print(torch.__version__)"

>>> 1.6.0

pip install torch-scatter==latest+cpu -f 

pip install torch-sparse==latest+cpu -f

pip install torch-cluster==latest+cpu -f

pip install torch-spline-conv==latest+cpu -f

pip install torch-geometric

import torch

import torch_geometric

若無報錯,則說明安裝成功,若有報錯,首先檢查一下安裝步驟,是否做到版本一致。

另外cuda的版本有可能顯示的不一定是真實安裝的版本。建議將cuda全部刪除,然後再安裝乙個新的。

Python遊戲開發 Mac完美安裝pygame

pygame是python的2d遊戲開發庫,優點是秉承了python風格,簡單好用 最近在macos high sierra10.13.5上安裝pygame時遇到了一些問題 經過一晚上的折騰,終於找到了正確的解決方法,分享一下 通過brew安裝依賴包 brew install sdl smpeg s...

DB2學習文件之一 初步安裝

出處 安裝了 db2,並成功建立例項之後,便可以以例項所有者的身份 在這個例子中是 db2inst1 登入,然後呼叫 db2profile 指令碼來設定環境變數。可以使用 db2sampl 指令碼建立樣例資料庫。為了開啟 db2 control center,記得要以 root 使用者發出 xhos...

ansible之一安裝

安裝 ansible的宿主機只能是linux機器,因為我用的是mac所以省去了很多麻煩。ansible是用python開發的,所以要先安裝python,mac自帶python所以直接安裝ansible 根據官方文件來 sudo pip install ansible執行第一條ansible命令ans...