宇宙の渚を眺めるエンジニアのブログ

技術的な備忘録や、日々思ったこと、たまに宇宙関連について綴る

Ubuntu 16.04のCUDA driver インストール

cupy使おうとしたら、cuda driver 関連でエラーが出たので cuda driverを新しくした。自分用の備忘録。

背景

cupy 使用時に以下のようなエラー(抜粋)が出た

> from chainer import cuda
> xp = cuda.cupy
> xp.arange(25)

cupy/cuda/memory.pyx in cupy.cuda.memory.MemoryPool.malloc()
cupy/cuda/memory.pyx in cupy.cuda.memory.MemoryPool.malloc()
cupy/cuda/device.pyx in cupy.cuda.device.get_device_id()
cupy/cuda/runtime.pyx in cupy.cuda.runtime.getDevice()
cupy/cuda/runtime.pyx in cupy.cuda.runtime.check_status()
CUDARuntimeError: cudaErrorInsufficientDriver: CUDA driver version is insufficient for CUDA runtime version

CUDA driver version がおかしいと言っているので、ひとまず最新化を実施。

インストール手順

自分の環境は以下のとおり
- OS: Ubuntu 16.04 (Windows 10とのデュアルブート) - CUDA 10.0

  1. 既存のCUDA driverを削除
sudo apt-get purge nvidia-*
  1. インストールしたいパッケージの確認
> ubuntu-drivers devices

== /sys/devices/pci0000:00/0000:00:01.0/0000:01:00.0 ==
vendor   : NVIDIA Corporation
modalias : pci:v000010DEd00001B06sv000010DEsd00001B06bc03sc00i00
driver   : nvidia-390 - third-party free
driver   : nvidia-415 - third-party free recommended
driver   : nvidia-384 - third-party non-free
driver   : nvidia-410 - third-party non-free
driver   : nvidia-396 - third-party free
driver   : xserver-xorg-video-nouveau - distro free builtin

nvidia-415 をインストールすることに

  1. インストール
sudo apt install nvidia-415

インストール中に別のウィンドウ立ち上がり、セキュアブート機能OFFにするか聞いてくる(おそらくデュアルブートの場合)。

  • セキュアブート機能をオOFFにするに同意 → パスワード設定 → そのままインストール進む → インストール終了後reboot
  • reboot時の青色の画面で、MOKうんちゃらでセキュアブートをオフにできる(画像撮ればよかった)。上記で設定したパスワード使用

以上でエラーが解消された。