Cara mudah: Instal driver Nvidia, CUDA, CUDNN dan Tensorflow GPU di Ubuntu 18.04

9

Bagaimana saya bisa dengan mudah menjalankan gpu tensorflow?

Jonny
sumber

Jawaban:

13

Saya baru saja menginstal GPU Tensorflow pada Ubuntu 18.04. Ada banyak instruksi untuk itu, namun saya pikir cara tercepat dan termudah biasanya tidak digunakan dan saya ingin membaginya:

NVIDIA DRIVER:

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

CUDA:

Biasanya: "sudo apt install nvidia-cuda-toolkit" Namun ini menginstal versi 9.1, terlalu baru saat ini dan tensorflow tidak akan berjalan. Alih-alih mengunduh 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 Kemudian jalankan (Rincian tambahan pada baris pertama dapat ditemukan di: Bagaimana cara menemukan di baris pertama?) Saya menginstal CUDA 9 di 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:

daftar di pengembang nvidia https://developer.nvidia.com/cudnn Unduh 9,1 runtime & perpustakaan pengembang untuk 16,04 (File cuDNN v7.1.3 Perpustakaan Runtime untuk Ubuntu16.04 (Deb) & cuDNN v7.1.3 Perpustakaan Pengembang untuk Ubuntu16.04 ( Deb)) Buka file dengan manajer perangkat lunak dan instal. Periksa dengan:

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

Instal perpustakaan dan tensorflow:

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

Memeriksa:

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

Catatan: Mulai tensorflow atau lingkungan pengembangan Anda dari terminal, jika tidak bagi saya itu tidak memuat variabel PATH.

Jonny
sumber
apakah ada cara agar cuda 10.1 berfungsi?
j0h
di nvidia's wesite, saya tidak melihat paket cudnn untuk versi 9.1. 9.0,9.2,10.0,10.1 adalah semua yang saya lihat, oh, dan versi sumber
j0h
2

Hanya untuk memperluas jawaban Jonny: ketika mengikuti metode Jonny untuk menginstal CUDA, saya harus memilih "Tidak" untuk "Instal NVIDIA Accelerated Graphics Driver untuk Linux-x86_64 384.81?" (ya atau default untuk yang lainnya). Jika tidak, instalasi driver CUDA akan gagal mengatakan "ERROR: Modul kernel NVIDIA 'nvidia-drm' tampaknya sudah dimuat di kernel Anda"

Kemudian Anda dapat menguji instalasi CUDA Anda dengan menjalankan sampel.

Pertama instal dependensi mereka:

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

Ini tidak akan dikompilasi dengan kompiler baru di Ubuntu 18, jadi Anda harus menurunkan versi ke 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

Kompilasi sampel dengan menavigasi ke:

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

Dan kompilasi menggunakan:

make -k

Jalankan salah satu sampel seperti:

./nbody
Robert Corvus
sumber