TensorFlow 1.7以上を使うにはCUDAの9.0以上を利用する必要があったのでCUDAのバージョンを8.0から9.0にアップグレードしました
【My環境】
ノートPC: ALIENWARE 15 R13
OS: Ubuntu 16.04 LTS(Windowsマシンでデュアルブート)
CUDA: 8.0
バージョンの確認
$ nvcc -V
バージョン8.0を使ってますね
nvcc: NVIDIA (R) Cuda compiler driver
Copyright (c) 2005-2016 NVIDIA Corporation
Built on Tue_Jan_10_13:22:03_CST_2017
Cuda compilation tools, release 8.0, V8.0.61
CUDA
NVIDIAの公式サイトからCUDA9.0のインストールしていきます
https://developer.nvidia.com/cuda-toolkit-archive
CUDA Toolkit 9.0を選択
以下の項目を指定して選択
- Operating System: Linux
- Architecture: x86_64
- Distribution: Ubuntu
- Version: 16.04
- Install Type: deb(network)
Downloadボタンをを押してファイルをダウンロード
debパッケージのインストール
$ sudo dpkg -i cuda-repo-ubuntu1604_9.0.176-1_amd64.deb
$ sudo apt-key adv --fetch-keys http://developer.download.nvidia.com/compute/cuda/repos/ubuntu1604/x86_64/7fa2af80.pub
$ sudo apt-get update
$ sudo apt-get install cuda-9-0
Pathの確認
$ echo $PATH
cuda-8.0がありますね
/usr/local/cuda-8.0/bin
cuDNN
cuDNNのダウンロードサイト(https://developer.nvidia.com/rdp/cudnn-download) から以下の3つをダウンロード
(下記3つは現在の最新バージョンに読み替えてください)
- cuDNN v7.0.5 Runtime Library for Ubuntu16.04 (Deb)
- cuDNN v7.0.5 Developer Library for Ubuntu16.04 (Deb)
- cuDNN v7.0.5 Code Samples and User Guide for Ubuntu16.04 (Deb)
ダウンロードしたファイルを以下のコマンドを実行してcuDNNのインストールを開始します
$ sudo dpkg -i libcudnn7_7.0.5.15-1+cuda9.0_amd64.deb
$ sudo dpkg -i libcudnn7-dev_7.0.5.15-1+cuda9.0_amd64.deb
$ sudo dpkg -i libcudnn7-doc_7.0.5.15-1+cuda9.0_amd64.deb
bashrcを編集
$ vim ~/.bashrc
Before
# NVIDIA CUDA Toolkit
export PATH=/usr/local/cuda-8.0/bin:$PATH
export LD_LIBRARY_PATH=/usr/local/cuda-8.0/lib64:$LD_LIBRARY_PATH
After
編集して保存
# NVIDIA CUDA Toolkit
export PATH=/usr/local/cuda-9.0/bin:$PATH
export LD_LIBRARY_PATH=/usr/local/cuda-9.0/lib64:$LD_LIBRARY_PATH
再読み込み
$ source ~/.bashrc
PATHの確認
$ echo $PATH
cuda-9.0発見
/usr/local/cuda-9.0/bin
バージョンを確認
$ nvcc -V
Hello, CUDA 9.0!
nvcc: NVIDIA (R) Cuda compiler driver
Copyright (c) 2005-2017 NVIDIA Corporation
Built on Fri_Sep__1_21:08:03_CDT_2017
Cuda compilation tools, release 9.0, V9.0.176