Kolay yol: Ubuntu 18.04'e Nvidia sürücülerini, CUDA, CUDNN ve Tensorflow GPU'yu yükleyin

9

Tensorflow gpu'yu nasıl kolayca çalıştırabilirim?

18.04 cuda gpu

— Jonny
kaynak

13

Ubuntu 18.04'e Tensorflow GPU'yu yeni yükledim. Bunun için çok fazla talimat var, ancak en hızlı ve en kolay yolun genellikle kullanılmadığını düşünüyorum ve paylaşmak istiyorum:

NVIDIA SÜRÜCÜ:

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

CUDA:

Normalde: "sudo apt install nvidia-cuda-toolkit" Ancak bu, 9.1 sürümünü yüklüyor, şu anda çok yeni ve tensorflow çalışmayacak. Bunun yerine CUDA 9.0'ı indirin: https://developer.nvidia.com/cuda-90-download-archive?target_os=Linux&target_arch=x86_64&target_distro=Ubuntu&target_version=1704&target_type=runfilelocal O zaman çalıştırın (İlk satırda Ek Ayrıntılar olabilir: İlk satırda nasıl olabilir: CUDA 9'u Ubuntu 17.10 üzerine yükledim ):

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 geliştiricilerine kaydolun https://developer.nvidia.com/cudnn 16.04 için 9.1 çalışma zamanı ve geliştirici kütüphanesini indirin (Dosyalar cuDNN v7.1.3 Ubuntu16.04 için Çalıştırma Kütüphanesi (Deb) & cuDNN v7.1.3 Ubuntu16.04 için Geliştirici Kitaplığı ( Deb)) Dosyaları yazılım yöneticisiyle açın ve yükleyin. Şununla kontrol et:

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

Kitaplıkları ve tensorflow'u yükleyin:

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

Kontrol:

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

Not: tensorflow'u veya geliştirme ortamınızı terminalden başlatın, aksi takdirde benim için PATH değişkenlerini yüklemez.

— Jonny
kaynak

cuda 10.1'in çalışmasını sağlamak için yine de var mı?

— j0h

nvidia'nın yerinde, 9.1 sürümü için herhangi bir cudnn paketi görmedim. 9.0,9.2,10.0,10.1 tüm gördüğüm, oh, ve bir kaynak versiyonu

— j0h

2

Jonny'nin cevabını genişletmek için: Jonny'nun CUDA'yı yükleme yöntemini izlerken, "Linux-x86_64 384.81 için NVIDIA Hızlandırılmış Grafik Sürücüsünü Kur?" (evet veya diğer her şey için varsayılan). Aksi takdirde, CUDA sürücü kurulumu "HATA: Bir NVIDIA çekirdek modülü 'nvidia-drm' çekirdeğinizde zaten yüklü görünüyor" diyerek başarısız olur.

Ardından örnekleri çalıştırarak CUDA kurulumunuzu test edebilirsiniz.

Önce bağımlılıklarını yükleyin:

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

Bunlar Ubuntu 18'deki yeni derleyicilerle derlenmeyeceğinden, gcc / g ++ 6 sürümüne geçmeniz gerekir:

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

Örnekleri şuraya giderek derleyin:

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

Ve kullanarak derleyin:

make -k

Aşağıdaki örneklerden birini çalıştırın:

./nbody

— Robert Corvus
kaynak