更改Colab的CUDA以及cudnn

Colab的CUDA版本是11.1，而我需要10.0的CUDA版本，更改CUDA版本过程如下：

1. 首先查看当前Colab的CUDA版本

!nvcc --version

2. 卸载当前Colab的CUDA

#Uninstall the current CUDA version
!apt-get --purge remove cuda nvidia* libnvidia-*
!dpkg -l | grep cuda- | awk '{print $2}' | xargs -n1 dpkg --purge
!apt-get remove cuda-*
!apt autoremove
!apt-get update

3. 去https://developer.nvidia.com/cuda-toolkit-archive下载你需要的CUDA版本，并安装；可以先看一下Colab的系统版本，应该都是Ubuntu 18.04；

!lsb_release -a

我的是这样的： 

 

然后知道系统版本，就可以去找对应的CUDA下载链接了，过程如下：

#Download CUDA 10.0
!wget  --no-clobber https://developer.nvidia.com/compute/cuda/10.0/Prod/local_installers/cuda-repo-ubuntu1804-10-0-local-10.0.130-410.48_1.0-1_amd64
#install CUDA kit dpkg
!dpkg -i cuda-repo-ubuntu1804-10-0-local-10.0.130-410.48_1.0-1_amd64
!sudo apt-key add /var/cuda-repo-10-0-local-10.0.130-410.48/7fa2af80.pub
!apt-get update
!apt-get install **cuda-10-0**

!apt-get install **cuda-10-0**是强制系统安装CUDA10.0，一定要记得加**，否则系统将会安装CUDA的最新版本！！！！！

4. 再次查看CUDA版本，检查一下是否安装成功

!nvcc --version

5. CUDA安装成功之后，再安装对应版本的cudnn，首先到https://developer.nvidia.com/rdp/cudnn-archive，找到要下载的cudnn的版本，注意CUDA10.0需要下载对应的三个并进行安装，此处我下载的是7.5.0版本

 注意，先检查一下路径下是否已经存在下载好的这三个文件，没有的话检查一下下载完之后的三个文件在哪，然后在其存在的路径下再进行安装：

!sudo dpkg -i libcudnn7_7.5.0.56-1+cuda10.0_amd64.deb
!sudo dpkg -i libcudnn7-dev_7.5.0.56-1+cuda10.0_amd64.deb
!sudo dpkg -i libcudnn7-doc_7.5.0.56-1+cuda10.0_amd64.deb

6. 将下载好的cudnn，合并到对应的cuda文件里

!sudo cp /usr/include/cudnn.h /usr/local/cuda/include
!sudo chmod a+x /usr/local/cuda/include/cudnn.h

 7.查看一下cudnn是否安装成功

!cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2

如果显示如下则成功：

参考链接：

https://aconcaguasci.blogspot.com/2019/12/setting-up-cuda-100-for-mxnet-on-google.html