Ubuntu 16.04安装N卡驱动、cuda和tensorfl

2017-11-17  本文已影响0人  佩鸿PH

安装驱动

最开始在英伟达官网下载了官方驱动,安装之后无法登录系统,在登录界面反复循环,用cuda里的驱动也出现了同样的问题。最后解决办法是把驱动卸载之后,通过命令行在线安装驱动。
卸载驱动:

sudo nvidia-uninstall

在线安装:

sudo apt-add-repository  ppa:graphics-drivers/ppa
sudo apt-get update
sudo apt-get install nvidia-384 #写这篇文章时,最新的N卡驱动是384,具体驱动版本号可以到官网查

安装完之后,在终端输入

nvidia-smi

出现下图,说明显卡安装成功:


安装cuda

在英伟达官网下载cuda安装包(.run文件),在终端中切换到cuda安装包目录,直接sudo运行安装包即可。注意在问你是否要安装显卡驱动的时候,选择no。
安装之后,添加环境变量:

gedit ~/.bashrc

在文件最后添加如下几行:

export PATH="/usr/local/cuda/bin:$PATH"
export CUDA_HOME="/usr/local/cuda"
export LD_LIBRARY_PATH="/usr/local/cuda/lib64:/usr/local/cuda/extras/CUPTI/lib64:$LD_LIBRARY_PATH"

安装tensorflow

安装之前需要安装libcupti-dev库,

sudo apt-get install libcupti-dev

安装tensorflow的方法很多,最简单的方法是用pip(或者用conda命令,如果你安装了anaconda的话)安装:

pip install tensorflow      # Python 2.7; CPU support (no GPU support)
pip3 install tensorflow     # Python 3.n; CPU support (no GPU support)
pip install tensorflow-gpu  # Python 2.7;  GPU support
pip3 install tensorflow-gpu # Python 3.n; GPU support 

亲测,下载速度太慢.下面介绍anaconda的安装方法。需要先安装上anaconda,里面包含了python以及大部分常用的科学计算工具包,例如numpy、pandas、matplotlib等。然后通过如下指令安装:

conda create -n tensorflow python=2.7 # or python=3.3, etc.
source activate tensorflow
pip install --ignore-installed --upgrade tfBinaryURL

tfBinaryURL是tensorflow二进制文件链接,不同版本的链接不同,例如安装python2.7 GPU版本:

pip install --ignore-installed --upgrade https://storage.googleapis.com/tensorflow/linux/gpu/tensorflow_gpu-1.4.0-cp27-none-linux_x86_64.whl

其他版本链接地址请参考:https://www.tensorflow.org/install/install_linux#the_url_of_the_tensorflow_python_package(要翻墙)
验证安装是否成功,在python命令行中输入如下代码:

import tensorflow as tf
hello = tf.constant('Hello, TensorFlow!')
sess = tf.Session()
print(sess.run(hello))

如果输出Hello, TensorFlow!,说明安装成功。如果报错,请参考如下链接:
https://www.tensorflow.org/install/install_linux#common_installation_problems

参考资料:

[1] https://www.tensorflow.org/install/install_linux

上一篇下一篇

猜你喜欢

热点阅读