簡単な方法：Ubuntu 18.04にNvidiaドライバー、CUDA、CUDNN、Tensorflow GPUをインストールする

9

tensorflow gpuを簡単に実行するにはどうすればよいですか？

18.04 cuda gpu

— ジョニー
ソース

13

Ubuntu 18.04にTensorflow GPUをインストールしました。手順はたくさんありますが、通常、最も速くて簡単な方法は使用されていないと思います。共有したいと思います。

NVIDIAドライバー：

ubuntu-drivers devices
sudo ubuntu-drivers autoinstall
nvidia-smi

CUDA：

通常：「sudo apt install nvidia-cuda-toolkit」ただし、これによりバージョン9.1がインストールされますが、現時点では新しすぎてテンソルフローは実行されません。代わりにダウンロードCUDA 9.0：https://developer.nvidia.com/cuda-90-download-archive?target_os=Linux&target_arch=x86_64&target_distro=Ubuntu&target_version=1704&target_type=runfilelocal 次に実行します（追加の詳細については、最初の行にで見つけることができます：どのようにすることができます私はCUDA 9をUbuntu 17.10にインストールします）：

sudo sh cuda_9.0.176_384.81_linux.run --override
sudo ln -s /usr/bin/gcc-6 /usr/local/cuda/bin/gcc
sudo ln -s /usr/bin/g++-6 /usr/local/cuda/bin/g++
cd ~
sudo nano .bashrc

add at the end of the file:
export PATH=/usr/local/cuda-9.0/bin${PATH:+:${PATH}}
export LD_LIBRARY_PATH=/usr/local/cuda-9.0/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}
CTL+X to save and exit

CuDNN：

nvidia開発者で登録https://developer.nvidia.com/cudnn 16.04用の9.1ランタイムと開発者ライブラリをダウンロード（Ubuntu16.04用のファイルcuDNN v7.1.3ランタイムライブラリ（Deb）＆Ubuntu16.04用のcuDNN v7.1.3開発者ライブラリ（ Deb））ソフトウェアマネージャーでファイルを開き、インストールします。確認する：

cat /usr/include/x86_64-linux-gnu/cudnn_v*.h | grep CUDNN_MAJOR -A 2

ライブラリとテンソルフローをインストールします。

sudo apt-get install libcupti-dev
pip3 install tensorflow-gpu

小切手：

in tensorflow check for GPU support 
sess = tf.Session(config=tf.ConfigProto(log_device_placement=True))

注：ターミナルからtensorflowまたは開発環境を起動します。そうしないと、PATH変数が読み込まれません。

— ジョニー
ソース

とにかくcuda 10.1を動作させる方法はありますか？

— j0h

nvidiaのウェブサイトで、バージョン9.1のcudnnパッケージを見つけませんでした。9.0、9.2、10.0、10.1は私が見たすべてで、ソースバージョンです

— -j0h

2

Jonnyの答えを拡張するために：CUDAをインストールするためのJonnyの方法に従うとき、「Install NVIDIA Accelerated Graphics Driver for Linux-x86_64 384.81？」で「いいえ」を選択する必要がありました。（はい、またはその他すべてのデフォルト）。そうしないと、CUDAドライバーのインストールが失敗し、「エラー：NVIDIAカーネルモジュール 'nvidia-drm'がすでにカーネルにロードされているようです」と表示されます。

次に、サンプルを実行して、CUDAインストールをテストできます。

まず、依存関係をインストールします。

sudo apt-get install freeglut3-dev build-essential libx11-dev libxmu-dev libxi-dev libgl1-mesa-glx libglu1-mesa libglu1-mesa-dev libglfw3-dev libgles2-mesa-dev

これらはUbuntu 18の新しいコンパイラーではコンパイルできないため、gcc / g ++ 6にダウングレードする必要があります。

sudo apt-get install gcc-6 g++-6
sudo update-alternatives --install /usr/bin/g++ g++ /usr/bin/g++-6 10
sudo update-alternatives --install /usr/bin/gcc gcc /usr/bin/gcc-6 10

次の場所に移動してサンプルをコンパイルします。

~/NVIDIA_CUDA-9.0_Samples/bin/x86_64/linux/release

そして、以下を使用してコンパイルします。

make -k

次のようなサンプルの1つを実行します。

./nbody

— ロバート・コーバス
ソース