最近準(zhǔn)備復(fù)現(xiàn)一下 KDD-20 Towards Deeper Graph Neural Networks 的代碼，順便學(xué)習(xí)一下 GCN 最新的一些進(jìn)展。

GCN 的代碼通常需要安裝 Pytorch on Graph, 按照其 官方指導(dǎo), 發(fā)現(xiàn)無法兼容當(dāng)前所使用的服務(wù)器上的 pytorch1.3 和 cuda10.0 環(huán)境, 需要升級 cuda 和 pytorch 的版本。

于是準(zhǔn)備重新安裝一個虛擬環(huán)境，使用上 cuda 10.2 和 pytorch1.6。

1. 安裝 Cuda10.2 以及對應(yīng)版本的 Cudnn

安裝 cuda 10.2，

wget http://developer.download.nvidia.com/compute/cuda/10.2/Prod/local_installers/cuda_10.2.89_440.33.01_linux.run
sudo sh cuda_10.2.89_440.33.01_linux.run

然后安裝對應(yīng)版本的 cudnn

但由于當(dāng)前機(jī)器上已經(jīng)安裝了 cuda 10.0，因此想要找一些能夠切換的方法，不刪去當(dāng)前的 cuda10.0, 發(fā)現(xiàn)可以使用軟鏈接的方式進(jìn)行切換

# 切換為 cuda 8.0 版本
rm -rf /usr/local/cuda #刪除之前創(chuàng)建的軟鏈接
sudo ln -s /usr/local/cuda-8.0/ /usr/local/cuda/
nvcc --version #查看當(dāng)前 cuda 版本

 nvcc: NVIDIA (R) Cuda compiler driver
 Copyright (c) 2005-2016 NVIDIA Corporation
 Built on Mon_Jan_23_12:24:11_CST_2017
 Cuda compilation tools, release 8.0, V8.0.62

# cuda8.0 切換到 cuda9.0 
rm -rf /usr/local/cuda
sudo ln -s /usr/local/cuda-9.0/ /usr/local/cuda/
nvcc --version

安裝 cuda10.2，這里只安裝 Cuda Toolkit 10.2, 不安裝 driver, 因為之前 Cuda10.0 已經(jīng)裝過 driver 了。

然后嘗試安裝 對應(yīng)版本的 cudnn, 在linux上

使用

wget "https://developer.nvidia.com/compute/machine-learning/cudnn/secure/8.0.3.33/10.2_20200825/cudnn-10.2-linux-x64-v8.0.3.33.tgz"

會導(dǎo)致一直顯示 403 Forbid 錯誤，但用 windows 瀏覽器可以訪問成功，用 windows 下載，但后綴名變?yōu)榱?".solitairetheme8",

mv cudnn-10.2-linux-x64-v8.0.3.33.solitairetheme8 cudnn-10.2-linux-x64-v8.0.3.33.tgz

接下來安裝 cudnn，注意到，現(xiàn)在有多個版本的 cuda 了，不要再使用軟鏈接 /usr/local/cuda/， 而應(yīng)該使用對應(yīng)版本的 cuda 地址。

使用 cd /usr/local/ 可以看到當(dāng)前安裝的 cuda 的情況， 我這里是：

使用如下命令完成 cudnn 的安裝：

tar -zxvf cudnn-10.2-linux-x64-v8.0.3.33.tgz
sudo cp cuda/include/cudnn.h /usr/local/cuda-10.2/include/
sudo cp cuda/lib64/libcudnn* /usr/local/cuda-10.2/lib64/
sudo chmod a+r /usr/local/cuda-10.2/include/cudnn.h
sudo chmod a+r /usr/local/cuda-10.2/lib64/libcudnn*

從官網(wǎng)下載實在太慢，也可以改為設(shè)置為 清華源

# 首先執(zhí)行如下幾條命令更換清華鏡像源conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/msys2/conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge/conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/conda config --set show_channel_urls yes

安裝結(jié)果：

突然發(fā)現(xiàn)，conda 也會安裝一個 cudatoolkit=10.2, 但這個和自己手動安裝的還是有很大的不同的

安裝 pytorch1.7 一直沒有成功，最新版的 pytorch1.7 需要從國外官網(wǎng)源頭直接進(jìn)行下載，速度太慢，一直失敗。
所以嘗試安裝 pytroch 1.5，

conda install pytorch=1.5

conda install torchvision

然后再按照 PoG 官網(wǎng)安裝相應(yīng)的包：

pip install torch-scatter
pip install torch-sparse
pip install torch-cluster
pip install torch-spline-conv
pip install torch-geometric

最后終于成功完成了安裝。

以上就是本文的全部內(nèi)容，希望對大家的學(xué)習(xí)有所幫助，也希望大家多多支持腳本之家。