Cách dễ dàng: Cài đặt trình điều khiển Nvidia, CUDA, CUDNN và GPU Tensorflow trên Ubuntu 18.04

9

Làm thế nào tôi có thể dễ dàng có được gpu dòng chảy?

18.04 cuda gpu

— Jonny
nguồn

13

Tôi vừa cài đặt GPU Tensorflow trên Ubuntu 18.04. Có rất nhiều hướng dẫn cho nó, tuy nhiên tôi nghĩ cách nhanh nhất và dễ nhất thường không được sử dụng và tôi muốn chia sẻ nó:

NVIDIA DRIVER:

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

CUDA:

Thông thường: "sudo apt install nvidia-cuda-toolkit" Tuy nhiên, bản cài đặt này phiên bản 9.1, quá mới tại thời điểm hiện tại và tenorflow sẽ không chạy. Thay vì tải 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 Sau đó chạy (Chi tiết bổ sung vào dòng đầu tiên có thể được tìm thấy tại địa chỉ: Làm thế nào có thể Tôi cài đặt CUDA 9 trên 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:

đăng ký tại nvidia developers https://developer.nvidia.com/cudnn Tải xuống thư viện nhà phát triển và thời gian chạy 9.1 cho 16.04 (Tệp cuDNN v7.1.3 Thư viện thời gian chạy cho Ubuntu16.04 (Deb) & cuDNN v7.1.3 Thư viện dành cho nhà phát triển cho Ubuntu16.04 ( Deb)) Mở tệp bằng trình quản lý phần mềm và cài đặt chúng. Kiểm tra với:

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

Cài đặt thư viện và tenorflow:

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

Kiểm tra:

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

Lưu ý: Bắt đầu tenorflow hoặc môi trường phát triển của bạn từ thiết bị đầu cuối, nếu không, đối với tôi, nó không tải các biến PATH.

— Jonny
nguồn

Có cách nào để cuda 10.1 hoạt động không?

— j0h

tại wesite của nvidia, tôi không thấy gói cudnn nào cho phiên bản 9.1. 9.0,9.2,10.0,10.1 là tất cả những gì tôi thấy, ồ, và một phiên bản nguồn

— j0h

2

Chỉ để mở rộng câu trả lời của Jonny: khi làm theo phương pháp của Jonny để cài đặt CUDA, tôi đã phải chọn "Không" cho "Cài đặt Trình điều khiển đồ họa tăng tốc NVIDIA cho Linux-x86_64 384.81?" (có hoặc mặc định cho mọi thứ khác). Nếu không, cài đặt trình điều khiển CUDA sẽ không nói "LRI: Mô-đun hạt nhân NVIDIA 'nvidia-drm' dường như đã được tải trong kernel của bạn"

Sau đó, bạn có thể kiểm tra cài đặt CUDA của mình bằng cách chạy các mẫu.

Đầu tiên cài đặt các phụ thuộc của họ:

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

Chúng sẽ không được biên dịch với các trình biên dịch mới trên Ubuntu 18, vì vậy bạn sẽ cần hạ cấp xuống 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

Biên dịch các mẫu bằng cách điều hướng đến:

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

Và biên dịch bằng cách sử dụng:

make -k

Chạy một trong các mẫu như:

./nbody

— Robert Corvus
nguồn