手動安裝這幾個包是總是出錯(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 表示成功