码迷,mamicode.com
首页 > 其他好文 > 详细

tensorflow的安装思路和技巧

时间:2018-03-05 13:46:11      阅读:193      评论:0      收藏:0      [点我收藏+]

标签:ubunt   出现   自己   load   公司   清华   好的   思路   query   

想了一下还是把tensorflow安装的过程整理一下吧,万一时间久了忘了呢。

1、准备工作

  1、更换源,好的软件源,直接决定你的安装速度。这里选择清华的。

  操作:进入:设置 --> 软件和更新 --> Ubuntu软件,打开下载自:选择 其他,选择清华。点击 选择服务器 确认,在弹出框选择,重新载入。

  2、更新软件包,防止有些工具没有安装上。

sudo apt update

  3、打开:设置 --> 软件和更新 --> 附加驱动

  选择NVIDIA公司专有的最新驱动,选择应用更改。时间较长,请耐心等待。

2、安装cuda

  下载文件,这里我选择run本地文件,进入文件所在目录,并运行,XXX代指文件名

sudo sh XXX.run

  安装注意,这时会出现很多信息,按ctr+c关闭信息,再按accept接受。

  接下来询问是否,安装驱动,输入n,不要安装。因为已经安装过。

  一路输入y,或者回车。

  进入/home/用户名/NVIDIA_CUDA-8.0_Samples/1_Utilities/deviceQuery,打开终端,输入

sudo make

  然后输入

./deviceQuery

  如果显示pass,说明cuda安装成功,如果显示说,cuda驱动不满足于cuda运行库,说明驱动太旧,或者cuda太新。参照修改。

3、配置cudnn软链接

cd  /home/***(自己的用户名)/Desktop/###(这个命令意思是找到刚刚我们用U盘传过来的文件)
tar xvzf cudnn-8.0-linux-x64-v5.1-ga.tgz###(解压这个文件)
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 /usr/local/cuda/lib64/libcudnn*

4、安装tensorflow  

在配置完cudnn之后,可以直接在终端输入

sudo pip install tensorflow==1.5.0

或者安装CPU版本

sudo pip install tensorflow==1.5.0

如果网速不好的话,也可以自己在https://github.com/tensorflow/tensorflow下载whl文件,往下翻,在installation下面,然后使用

 

sudo pip install xxx.whl

  

tensorflow的安装思路和技巧

标签:ubunt   出现   自己   load   公司   清华   好的   思路   query   

原文地址:https://www.cnblogs.com/smartmsl/p/8508332.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!