Ubuntu 16.04安装NVIDIA GTX 1070和TensorFlow指南

最新推荐文章于 2024-09-14 09:58:54 发布

转载最新推荐文章于 2024-09-14 09:58:54 发布 · 3.8k 阅读

标签

#tensorflow #linux #ubuntu #CUDA

ubuntu 同时被 3 个专栏收录

25 篇文章

订阅专栏

CUDA

17 篇文章

订阅专栏

tensorflow

3 篇文章

订阅专栏

本文详细介绍如何在Ubuntu16.04.1上安装NVIDIA GTX1070显卡驱动，并配置CUDA8.0RC、cuDNN5.1及TensorFlow0.10.0 GPU版本，适用于深度学习开发环境的快速部署。

0. 准备工作
硬件：NVIDIA GTX 1070显卡一块
软件：Ubuntu 16.04.1/CUDA 8.0 RC/cuDNN 5.1/TensorFlow 0.10.0
1. 安装Ubuntu 16.04.1
2. 安装NVIDIA 365.35 驱动程序

sudo
add-apt-repository ppa:graphics-drivers/ppa
sudo
apt-get update
sudo
apt-get install
nvidia-367
sudo
reboot

3. 安装CUDA 8.0 RC，需要使用runfile安装方案，不要选择deb方案。

sudo
sh cuda_8.0.27_linux.run

备注：在安装过程中非常关键的地方是会提示你是否安装361这个低版本的驱动：

Install NVIDIA Accelerated Graphics Driver
for Linux-x86_64 361.62?

答案必须是n，因为之前我们已经安装过高版本的GTX1070驱动了。
4. 使用nvidia-smi命令测试显卡驱动和CUDA是否安装成功

Thu Aug 4 01:19:40 2016
+-----------------------------------------------------------------------------+
| NVIDIA-SMI 367.35 Driver Version: 367.35 |
|-------------------------------+----------------------+----------------------+
| 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 1070 Off | 0000:01:00.0 Off | N/A
|
| 0% 40C P8 11W / 166W | 103MiB / 8112MiB | 0% Default |
+-------------------------------+----------------------+----------------------+

5. 安装cuDNN 5.1

tar
-zxf cudnn-8.0-linux-x64-v5.1.tgz
cd
cuda
sudo
cp lib64/* /usr/local/cuda/lib64/
sudo
cp include/* 
/usr/local/cuda/include/

6. 在.bashrc文件中添加和CUDA相关的环境变量

export
PATH=/usr/local/cuda-8.0/bin${PATH:+:${PATH}}
export
LD_LIBRARY_PATH=/usr/local/cuda-8.0/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}

7. 然后在命令行输入

source
~/.bashrc

8. 使用命令nvcc -V测试环境变量是否配置成功

$ nvcc -V
nvcc: NVIDIA (R) Cuda compiler driver
Copyright (c) 2005-2016 NVIDIA Corporation
Built on Sun_Sep__4_22:14:01_CDT_2016
Cuda compilation tools, release 8.0, V8.0.44