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. 下载你需要的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
参考链接:
https://aconcaguasci.blogspot.com/2019/12/setting-up-cuda-100-for-mxnet-on-google.html