web-dev-qa-db-ja.com

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

テンソルフローGPUを簡単に実行するにはどうすればよいですか？

18.04cudagpu
10
Jonny

Ubuntu 18.04にTensorflow GPUをインストールしました。それには多くの指示がありますが、私は通常、最速かつ最も簡単な方法は使用されていないと思うので、共有したいと思います：

NVIDIAドライバー：

ubuntu-drivers devices
Sudo ubuntu-drivers autoinstall
nvidia-smi

CUDA：

通常：「sudo apt install nvidia-cuda-toolkit」ただし、これはバージョン9.1をインストールするため、現時点ではあまりにも新しいため、tensorflowは実行されません。代わりに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 次に実行します（最初の行の追加の詳細は、次の場所にあります： Ubuntu 17.10にCUDA 9をインストールする方法 ）：

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ランタイムおよび開発者ライブラリをダウンロード（Files cuDNN v7.1.3 Runtime Library for Ubuntu16.04（Deb）＆Ubuntu16.04（Deb）用cuDNN v7.1.3 Developer Library）ソフトウェアマネージャーでファイルを開き、インストールします。確認する：

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

ライブラリとtensorflowをインストールします。

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変数が読み込まれません。

14
Jonny

Jonnyの答えを拡張するために、CUDAをインストールするJonnyの方法に従うとき、「Linux-x86_64 384.81用のNVIDIA Accelerated Graphics Driverをインストールしますか？」で「いいえ」を選択する必要がありました。 （はいまたは他のすべてのデフォルト）。そうしないと、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

次のようなサンプルのいずれかを実行します。

./nbody
2
Robert Corvus