因為需要跑一些別人的GCN的任務,所以就要安裝相關的包; 首先要注意cuda版本和pytorch中的對應關系,必須嚴格對應,比如我的機器上,cuda版本是10.1; pytorch的cuda版本也是10.1,之前其實是10.2,又重裝了一遍; pytorch ...
手動安裝這幾個包是總是出錯 OSError: WinError 找不到指定的模塊 ,是因為版本之間的沖突,可以使用命令自動尋找版本。 安裝步驟: 安裝pytorch 下載鏈接:https: download.pytorch.org whl torch stable.html 選擇對應的版本,包括torch版本 cpu cuda python不同版本 windows linux。 比如我現在的是to ...
2020-05-25 10:49 0 5749 推薦指數:
因為需要跑一些別人的GCN的任務,所以就要安裝相關的包; 首先要注意cuda版本和pytorch中的對應關系,必須嚴格對應,比如我的機器上,cuda版本是10.1; pytorch的cuda版本也是10.1,之前其實是10.2,又重裝了一遍; pytorch ...
(GNN) zhang@zhang:~$ pip install torch-scatter -f https://pytorch-geometric.com/whl/torch-1.8.0+cpu.html ...
1.從數據直接構建tensor x = torch.tensor([5.5,3]) 2.從已有的tensor構建一個tensor。這些方法會重用原來tensor的特征。 x = x.new_ones(5,3,dtype=torch.double) torch.randn_like(x ...
1. Pytorch 和 PyG Pytorch和PyG是深度學習的python環境,后者用於圖處理。其主要特點如下: ①需要與cuda(NVIDIA)配合,必須先裝載cuda。 ②cuda,pytorch,pytorch_geometric三者之間版本依賴精密,必須屬於同一套版本。 其安裝順序 ...
官方文檔:https://pytorch-geometric.readthedocs.io/en/latest/notes/installation.html 按照官方文檔,確保cuda和nvcc版本一致 我的運行如下所示: 然后安裝依賴,官方文檔中 ...
1、官網上查看自己的環境對應的torch_geometric版本 (1)github網址:https://github.com/rusty1s/pytorch_geometric (2)打開網站之后,能在網頁上看到 “Installation”,我用到的版本如下圖: 2、安裝 ...
直接用pip安裝一些包會失敗,比如使用 安裝torch_scatter會失敗 通用的解決辦法就是手動把包下下來,本地安裝。 對於torch_scatter,可以在https://pytorch-geometric.com/whl/torch-1.5.0.html下載對應版本,記得選擇 ...
1.如何構造一個稀疏矩陣呢? 輸出如下: 2.怎么獲得這個稀疏矩陣的indices呢? 至於為啥要coalesce我也不曉得,反正就記住唄。對了,忘了說了,上面有寫到,給一個torch tensor轉置用函數t()。 輸出 ...