ubuntu18.04 nvidia gpu驱动与tensorflow-gpu以 docker方式安装

最新推荐文章于 2024-04-26 13:44:34 发布

Geikon

最新推荐文章于 2024-04-26 13:44:34 发布

阅读量1.1k

点赞数 1

分类专栏： ubuntu系统使用文章标签： navidia 驱动安装 tensorflow -gpu

本文链接：https://blog.csdn.net/Geikon/article/details/94431563

版权

ubuntu系统使用专栏收录该内容

5 篇文章 0 订阅

订阅专栏

通过ubuntu标准库自动安装驱动

搜索系统当前支持的驱动（此方法可能无法安装最新版）

 $ ubuntu-drivers devices
== /sys/devices/pci0000:00/0000:00:01.0/0000:01:00.0 ==
modalias : pci:v000010DEd00001180sv00001458sd0000353Cbc03sc00i00
vendor   : NVIDIA Corporation
model    : GP106 [GeForce GTX 1060 6GB]
driver   : nvidia-304 - distro non-free
driver   : nvidia-340 - distro non-free
driver   : nvidia-390 - distro non-free recommended
driver   : xserver-xorg-video-nouveau - distro free builtin

== cpu-microcode.py ==
driver   : intel-microcode - distro free

命令行自动安装

sudo ubuntu-drivers autoinstall

或者可以指定可以选择的安装版本

sudo apt install nvidia-340

重启系统，之后验证安装是否成功

nvidia-smi

成功则返回类似如下的结果

+-----------------------------------------------------------------------------+
| NVIDIA-SMI 390.116                Driver Version: 390.116                   |
|-------------------------------+----------------------+----------------------+
| GPU  Name        Persistence-M| Bus-Id        Disp.A | Volatile Uncorr. ECC |
| Fan  Temp  Perf  Pwr:Usage/Cap|         Memory-Usage | GPU-Util  Compute M. |
|===============================+======================+======================|
|   0  GeForce GTX 106...  Off  | 00000000:01:00.0 Off |                  N/A |
| 35%   39C    P0    25W / 120W |     12MiB /  6078MiB |      0%      Default |
+-------------------------------+----------------------+----------------------+
                                                                               
+-----------------------------------------------------------------------------+
| Processes:                                                       GPU Memory |
|  GPU       PID   Type   Process name                             Usage      |
|=============================================================================|
|    0      2137      G   /usr/lib/xorg/Xorg                             8MiB |
|    0      3072      G   /usr/bin/gnome-shell                           1MiB |
+-----------------------------------------------------------------------------+

添加第三方源自动安装驱动

# 添加源
$ sudo add-apt-repository ppa:graphics-drivers/ppa
$ sudo apt update

查找推荐的驱动
$ ubuntu-drivers devices
== /sys/devices/pci0000:00/0000:00:01.0/0000:01:00.0 ==
modalias : pci:v000010DEd00001C03sv00001043sd000085ABbc03sc00i00
vendor   : NVIDIA Corporation
model    : GP106 [GeForce GTX 1060 6GB]
driver   : nvidia-driver-390 - third-party free
driver   : nvidia-driver-410 - third-party free recommended
driver   : nvidia-driver-396 - third-party free
driver   : xserver-xorg-video-nouveau - distro free builtin

自动安装
sudo ubuntu-drivers autoinstall

安装完重启，验证

手动安装驱动

到官方网站下载驱动，安装prerequisites, 禁用nouveau，关闭图形桌面，安装驱动，重启。

安装tensorflow-gpu

使用docker进行安装，确保docker已经安装好，运行如下命令安装nvidia-docker2

# Add the package repositories
curl -s -L https://nvidia.github.io/nvidia-docker/gpgkey | \
  sudo apt-key add -
distribution=$(. /etc/os-release;echo $ID$VERSION_ID)
curl -s -L https://nvidia.github.io/nvidia-docker/$distribution/nvidia-docker.list | \
  sudo tee /etc/apt/sources.list.d/nvidia-docker.list
sudo apt-get update

# Install nvidia-docker2 and reload the Docker daemon configuration
sudo apt-get install -y nvidia-docker2
sudo pkill -SIGHUP dockerd

验证nvidia-docker安装，并安装验证tensorflow-gpu

docker run --runtime=nvidia --rm nvidia/cuda nvidia-smi
docker run --runtime=nvidia -it --rm tensorflow/tensorflow:latest-gpu \
       python -c "import tensorflow as tf; tf.enable_eager_execution(); print(tf.reduce_sum(tf.random_normal([1000, 1000])))"