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

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

!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