首先在https://pytorch-geometric.com/whl/，查看對應版本的whl文件，這里進入torch-1.7.1+cpu，因為我們使用的是python3.6版本，所以選擇的是cp36（這是根據具體python版本來決定的）。然后下載，如下圖所示:

選擇對應的文件：

torch_scatter-2.0.7-cp36-cp36m-win_amd64.whl torch_sparse-0.6.9-cp36-cp36m-win_amd64.whl torch_spline_conv-1.2.1-cp36-cp36m-win_amd64.whl torch_cluster-1.5.9-cp36-cp36m-win_amd64.whl

將whl文件下載到電腦上，使用pip進行安裝，過程如下：

最后使用pip install torch-geometric命令安裝即可，下面展示部分截圖：

簡單測試一下torch_geometric：

import torch from torch_geometric.data import Dataedge_index = torch.tensor([[0,1,1,2],[1,0,2,1] ], dtype=torch.long) x = torch.tensor([[-1], [0], [1]], dtype=torch.float)data = Data(x=x, edge_index=edge_index)

這里我們創建了一個自定義的數據，輸出結果：

說明沒有問題了

總結

以上是生活随笔為你收集整理的【windows】安装pytorch1.7.1-cpu版本的geometric过程记录的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。