深度學習環境之Pytorch和torch_Geometric的搭建


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。

  1. 進入Pytorch官網(https://pytorch.org/),選擇對應的版本號(可向下兼容),這里我們選擇穩定版本1.10.2和cuda10.2
  2. 將圖示中Run this Command所提供的代碼在Anoconda激活的環境中運行(即activate phm2012 phm2012為環境名),安裝pytorch。

2.4 Pytorch_Geometri的安裝

同理,基於上述基礎,在Anoconda中進行PyG的導入:

  1. 進入PyG官網(https://pytorch-geometric.readthedocs.io/en/latest/notes/installation.html)
  2. 選擇版本號,在Anoconda激活的環境中安裝。

2.5 Pycharm中找到該環境

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


免責聲明!

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



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