LoginSignup
11
9

More than 5 years have passed since last update.

UbuntuにCUDAをインストール

Last updated at Posted at 2017-07-23

はじめに

そろそろ DeepLearning に手を出したくて CUDA を入れることに。先人が多く、既にUbuntuにCUDAを入れる記事が多く見つかるが、記事に従ってもすんなりとインストールに成功しなかった。筆者がインストールに成功した方法をまとめるので、苦労している誰かに役立てば幸いである。

目的

Ubuntu 16.04LTS (GeForce GTX 750搭載) に CUDA 8 をインストールする。

Download Ubuntu
Download CUDA Toolkit

CUDA Toolkit インストール方法

Ubuntu のインストールが既に完了しているとする。

1. 事前準備

nouveauが有効になっていると正常にインストールできないため、nouveauを無効にする。
CUDA Toolkit インストールマニュアルより、nouveauを無効にする方法を引用する。

  1. ファイル /etc/modprobe.d/blacklist-nouveau.conf を作成し、以下の文字列を書き込む
/etc/modprobe.d/blacklist-nouveau.conf
blacklist nouveau
options nouveau modeset=0

 
2. Initramfsを再構築する

$ sudo update-initramfs -u

 
3. 再起動

$ sudo reboot

 
4. BIOS設定でUEFI無効またはセキュアブートを無効にする

2. CUDA Toolkit インストール

  1. Download CUDA Toolkit から、 runfile(local) をダウンロード
  2. 2. ディスプレイマネージャが起動しているとインストールできないため、停止させる
$ sudo service lightdm stop

 
3. Ctrl+Alt+F1 でCUIモードログインをし、1. でダウンロードした runfile(local) を実行してインストール
4. Ubuntu 16.04へのCUDAインストール方法 より、パスの設定を行う。設定ファイルと設定方法を以下に引用する。

~/.bashrc
export PATH="/usr/local/cuda-8.0/bin:$PATH"
export LD_LIBRARY_PATH="/usr/local/cuda-8.0/lib64:$LD_LIBRARY_PATH"

 
5. 再起動

$ sudo reboot

 
6. GUIからログインできたらインストール完了。出来ていなかったら次に進む

3. NVidia GPUのドライバ再インストール

UbuntuにNvidia GPUのDriver + CUDAをInstallする(GTX1080対応版) の2. 3. 4. 5. 6.を行う。2. 3. 4. 5. 6. を引用し、編集したのを以下に示す
1. 古いドライバーの削除

$ sudo apt-get --purge remove nvidia-*

 
2. リポジトリの登録

$ sudo add-apt-repository ppa:graphics-drivers/ppa
$ sudo apt-get update

 
3. 搭載GPUに適したドライバーのインストール

$ sudo apt-get install nvidia-375

ドライバのバージョンは、Download Drivers で該当GPUを検索した際のバージョンになる。
 
4. 再起動

$ sudo reboot

 
5. 確認

$ nvidia-smi

特にエラーが出なければOK

11
9
0

Register as a new user and use Qiita more conveniently

  1. You get articles that match your needs
  2. You can efficiently read back useful information
  3. You can use dark theme
What you can do with signing up
11
9