联想Y720,Ubuntu下搭建TensorFlow环境(GPU版,Python2.7)

本机配置参考第一篇文章https://blog.csdn.net/qq_23612831/article/details/88289943

1、先安装和更新各个包

sudo apt-get update
sudo apt-get install python-pip python-dev
sudo pip install --upgrade pip

2、安装NVIDIA驱动

NVIDIA驱动过程安装略过
附:CUDA与NVIDIA驱动对应关系:

CUDA9.19.08.07.57.0
NVIDIA387.xx384.xx367.4x352.xx346.xx

本机为NVIDIA 384.130,故可以安装CUDA9.0
附:降gcc版本方法(CUDA8.0支持gcc5.0以下,CUDA9.0支持gcc6.0以下)
降gcc到5.0以下方法:
sudo apt-get install g++ -4.9

3、依赖库安装(CUDA9.0)

sudo apt-get install freeglut3-dev build-essential libxll-dev libxmu-dev
sudo apt-get install libxi-dev libgll-mesa-glx libglul-mesa libglul-mesa-dev

4、下载CUDA9.0相关

下载CUDA9.0 (linux—x86_64—Ubuntu—16.04—deb)
下载Base Install 以及其几个patch

5、将安装包放在自己指定目录

安装sudo dpkg -i cuda-rep-ubutu1604-9-0xxxxx.deb(省略安装文件名字)
中途需要按要求输入key指令:sudo apt-key add /var/……(安装过程有提示,省略)
安装几个patchsudo dpkg -i xxxxx(省略安装包名称)

sudo apt-get update
sudo apt-get install cuda

此时,可见,在目录 usr/local 下,已经有cuda,cuda-9.0两个文件夹

6、环境配置

注:用户安环境设置: sudo gedit ~/.bashrc
让设置生效 :source ~/.bashrc
系统环境设置:sudo gedit /etc/profile
让设置生效 :source /etc/profile

本次安装,设置的是系统环境:sudo gedit /etc/profile
在打开的文件末尾添加:

export PATH = /usr/local/cuda-9.0/bin:$PATH
export CUDA_HOME = /usr/local/cuda-9.0
export LD_LIBRARY_PATH = /usr/local/cuda-9.0/lib64:$LD_LIBRARY_PATH
export LD_LIBRARY_PATH = /usr/local/cuda-9.0/extras/CUPT/lib64:$LD_LIBRARY_PATH 

保存退出以后,再次执行source /etc/profile

7、测试CUDA安装成功与否

方法1:

cd /usr/local/cuda9.0/samples/1_Utilities/deviceQuery
sudo make
./deviceQuery

若有GPU信息,则安装成功(最后会有一个Result = PASS)

方法2:

cd /usr/local/cuda9.0/samples/5_Simulations/fluidsGL
sudo make
./fluidsGL

如果最后出现一个动态的花里胡哨的界面,则证明CUDA安装成功

8、cuDnn 7.0 安装

下载:cudnn v7.0.5 for CUDA 9.0
cudnn v7.0.5 Library for linux
将下载的包放到自己指定的目录
在目录下运行:tar -xvzf cudnn-9.0-linux-x64-v7.solitairetheme8(后面为自己下载的包名)

sudo cp cuda/include/cudnn.h /usr/local/cuda/include
sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64
sudo chmod a+r /usr/local/cuda/include/cudnn.h
sudo chmod a+r /usr/local/cuda/lib64/libcudnn*

(注:将复制的目标目录的cuda更改为cuda-9.0也一样,若上述测试不成功,可以更改为cuda9.0,例:sudo cp cuda/include/cudnn.h /usr/local/cuda-9.0/include
测试安装成功与否:nvcc-V若结果显示相关信息,则表明测试成功。

9、安装TensorFlow

安装前,先安装:sudo apt-get install libcupti-dev
网上下载好TensorFlow对应版本,本次使用tensorflow_gpu_1.10_cp27_cp27mu_manylinuxl_x86_x64.whl
安装:sudo -H pip install xxxx.whl(xxx为对应安装包)
注:若提示pip是家目录的,则用sudo -H ,没提示,则可以不用 -H
测试:

python
import tensorflow as tf
hello = tf.constant('hello world')
sess = tf.session()
print = (sess.run(hello))

若结果打印出“hello world”,则表示安装成功。

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 3
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 3
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值