将Colab上的CUDA版本降级至9

在Google Colab上降级CUDA版本较为困难，因为Colab预装了特定版本的CUDA和cuDNN，并且它们会自动更新

首先，检查当前CUDA版本：

!nvcc --version

卸载当前CUDA版本：

!apt-get --purge remove cuda nvidia-cuda-toolkit
!dpkg -l | grep cuda- | awk '{print $2}' | xargs -n1 dpkg --purge
!apt-get --purge remove cuda-*
!apt autoremove
!apt-get update

安装CUDA 9：

!wget https://developer.nvidia.com/compute/cuda/9.0/Prod/local_installers/cuda-repo-ubuntu1804-9-0-local_9.0.176-1_amd64.deb
!dpkg -i cuda-repo-ubuntu1804-9-0-local_9.0.176-1_amd64.deb
!apt-key add /var/cuda-repo-9-0-local/7fa2af80.pub
!apt-get update
!apt-get install cuda-9.0

安装cuDNN 7（与CUDA 9兼容的版本）：

!wget https://developer.nvidia.com/compute/machine-learning/cudnn/secure/v7.6.5/prod/9.0_20191031/cudnn-9.0-linux-x64-v7.6.5.32.tgz
!tar -xvf cudnn-9.0-linux-x64-v7.6.5.32.tgz
!cp cuda/include/cudnn.h /usr/local/cuda/include
!cp cuda/lib64/libcudnn* /usr/local/cuda/lib64
!chmod a+r /usr/local/cuda/include/cudnn.h /usr/local/cuda/lib64/libcudnn*

更新环境变量：

import os
os.environ["CUDA_HOME"] = "/usr/local/cuda"
os.environ["PATH"] = "/usr/local/cuda/bin:" + os.environ["PATH"]
os.environ["LD_LIBRARY_PATH"] = "/usr/local/cuda/lib64:" + os.environ["LD_LIBRARY_PATH"]