安裝torch,torch_geometric,torch_sparse,torch_scatter


手動安裝這幾個包是總是出錯(OSError: [WinError 126] 找不到指定的模塊),是因為版本之間的沖突,可以使用命令自動尋找版本。

安裝步驟:

1)安裝pytorch

  下載鏈接:https://download.pytorch.org/whl/torch_stable.html 

  選擇對應的版本,包括torch版本、cpu/cuda、python不同版本、windows/linux。

  比如我現在的是torch=1.4.0、cu101、py36、linux版本   #箱安裝cuda版本的torch,可通過nvcc -v 查看已安裝的cuda版本

2)在命令行執行下述命令

  CUDA=cu92 #  cpu、cu92、cu100或cu101

  pip install torch-scatter==latest+${CUDA} -f https://pytorch-geometric.com/whl/torch-1.4.0.html

  pip install torch-sparse==latest+${CUDA} -f https://pytorch-geometric.com/whl/torch-1.4.0.html

  pip install torch-cluster==latest+${CUDA} -f https://pytorch-geometric.com/whl/torch-1.4.0.html

  pip install torch-spline-conv==latest+${CUDA} -f https://pytorch-geometric.com/whl/torch-1.4.0.html

  pip install torch-geometric

遇到的錯誤:

一、運行項目時from torch_geometric.utils import scatter_這一行報錯:ImportError: cannot import name 'scatter_'

  是因為torch_geometric版本不對,可以嘗試重新安裝低版本或高版本的torch_geometric:

    pip install torch_geometric==1.4.1

 

二、在ubuntu上重裝GPU版本,出現錯誤:libcublas.so.9.2: cannot open shared object file: No such file or directory

這個錯誤可能是如下問題導致的:

  • Cuda-9.2未安裝或安裝不正確
  • 未添加環境變量或添加完未更新

1. 首先查看是否安裝了cuda-9.2:(nvcc -v 或查看  /usr/local/cuda-9.2/lib64/ibcublas.so.9.2存在)。

2. 若已安裝檢查環境變量是否添加

sudo gedit .bashrc
在末尾加入:
export  PATH="$PATH:/usr/local/cuda-9.2/bin" export  LD_LIBRARY_PATH="$LD_LIBRARY_PATH:/usr/local/cuda-9.2/lib64"
更新環境變量
source .bashrc

  如果還是報錯,考慮卸載並重新安裝:

cd /usr/local/cuda/bin
sudo ./uninstall_cuda_9.2.pl

3. 安裝cuda-9.2

  下載鏈接: https://developer.nvidia.com/cuda-92-download-archive?target_os=Windows&target_arch=x86_64

  1)檢查環境是否滿足安裝cuda條件:

lspci | grep -i nvidia  #電腦中是否有支持cuda的GPU
uname -m && cat /etc/*release #linux系統是否支持cuda
#結果    x86_64 DISTRIB_ID=Ubuntu DISTRIB_RELEASE=16.04 ......
 gcc –version  #系統是否安裝了gcc
#結果  gcc (Ubuntu 5.4.0-6ubuntu1~16.04.5) 5.4.0 20160609 ......
#若未安裝請使用下列命令進行安裝: sudo apt-get  install  build-essential
sudo apt-get install linux-headers-$(uname -r) #可以安裝對應kernel版本的kernel header和package development 
#結果  ...... 升級了 0 個軟件包,新安裝了 0 個軟件包,要卸載 0 個軟件包,有 4 個軟件包未被升級。 表示系統里已經有了,不用重復安裝。

  2)采用runfile安裝cuda-9.2

  到cuda官網,根據系統信息下載cuda-9.2的  XXX.run文件

lsmod | grep nouveau #如果有輸出則代表nouveau正在加載。需要我們手動禁掉nouveau。

#Ubuntu的nouveau禁用方法:
#在/etc/modprobe.d中創建文件blacklist-nouveau.conf 輸入命令:
sudo vi /etc/modprobe.d/blacklist-nouveau.conf  
#利用vi編輯器編輯和保存文件) 在文件中輸入一下內容: 
blacklist nouveau options nouveau modeset=0 
sudo update-initramfs –u 
lsmod | grep nouveau #若無內容輸出,則禁用成功,若仍有內容輸出,請檢查操作,並重復上述操作。

  Ctrl+Alt+F1進入文本模式(命令行界面):

sudo sh cuda_9.1.85_387.26_linux.run --no-opengl-libs
#在提示 Install NVIDIA Accelerated Graphics Driver for Linux-x86_64 387.26? (y)es/(n)o/(q)uit: 時,選n,其他選y

 添加環境變量並更新(參考2.),並完成以下操作

sudo vim /etc/profile export #設置環境變量和動態鏈接庫
PATH=/usr/local/cuda/bin:$PATH #在文件尾部加入

sudo vim /etc/ld.so.conf.d/cuda.conf # 創建鏈接文件
/usr/local/cuda/lib64 #在文件尾部加入
sudo ldconfig 

#測試 cuda 的 Samples:
cd NVIDIA_CUDA-9.1_Samples
sudo make all -j4
cd bin/x86_64/linux/release
./deviceQuery

#結果 ....deviceQuery...Device0 = GeForce GT 635M Result = PASS 表示成功

  參考:https://www.jianshu.com/p/00c37b09f0f3

     https://blog.csdn.net/qq_39521554/article/details/82829886


免責聲明!

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



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