忘備録 colab CUDAバージョン固定

今度はcolabのCUDAが11へバージョンアップされたようでLightGBMのGPU版が動かなくなってしまった。
CUDAのバージョンを10へダウンさせるコード。↓

# https://zenn.dev/takeshita/articles/a02402e59d72a7
!wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu1804/x86_64/cuda-ubuntu1804.pin
!sudo mv cuda-ubuntu1804.pin /etc/apt/preferences.d/cuda-repository-pin-600
!sudo apt-key adv --fetch-keys https://developer.download.nvidia.com/compute/cuda/repos/ubuntu1804/x86_64/7fa2af80.pub
!sudo add-apt-repository "deb https://developer.download.nvidia.com/compute/cuda/repos/ubuntu1804/x86_64/ /"
!sudo apt-get update
!sudo apt-get -y installnvidia-driver-418
!sudo apt-get -y install cuda-10-2  #←デフォルトはcudaになってるので10.2をインストールするように書きかえる

!nvcc --version
!nvidia-smi