在Google colab Colaboratory上，安装CUDA和GPU版本的MXnet

版权声明：版权归本作者所有，如需转载请务必注明出处及作者 https://blog.csdn.net/qq_35654046/article/details/87601561

首先得查看一下操作系统的版本

!sudo lsb_release -a

Google colab 目前用的是Ubuntu 18.04

到官网上找到CUDA对应的版本

https://developer.nvidia.com/cuda-downloads?target_os=Linux&target_arch=x86_64&target_distro=Ubuntu&target_version=1804&target_type=deblocal

安装CUDA
 

!apt-get update && sudo apt-get install -y build-essential git libgfortran3
!wget 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
!sudo dpkg -i cuda-repo-ubuntu1804-10-0-local-10.0.130-410.48_1.0-1_amd64.deb
!sudo apt-key add /var/cuda-repo-<version>/7fa2af80.pub
!sudo apt-get update
!sudo apt-get install cuda

查看CUDA版本

!cat /usr/local/cuda/version.txt

安装MXnet

!pip install mxnet-cu100
import mxnet as mx