更改Colab的CUDA

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

上一篇:Google Colab导入GitHub项目


下一篇:Google在线深度学习神器Colab——科研角色