Ubuntu16.04+gtx1070显卡+安装TensorFlowGPU版

2023-11-06 07:59

本文主要是介绍Ubuntu16.04+gtx1070显卡+安装TensorFlowGPU版,希望对大家解决编程问题提供一定的参考价值,需要的开发者们随着小编来一起学习吧!

Ubuntu16.04+gtx1070显卡+安装TensorFlowGPU版

首先我先推荐大家去看这2篇我的文章,第一篇讲,怎么制作Ubuntu16的U盘安装,里面没讲怎么安装Ubuntu,我们用的最简单的方法,直接没有保存电脑上面的数据,盘也懒得分了,网上有教程,这里就不细讲了。

大概流程是,开机显示屏提示按F12 你不停的按,进入装机界面,选第5个:USB-uefi .....

接着选第2个,直接按Ubuntu。后面忘了,记得有个选择安装模式,我选第一个,就省掉了分盘操作。

https://mp.csdn.net/postedit/79722615


第二篇讲了,针对我的1070显卡,怎么选择合适的驱动,怎么安装。可能一般的用户的话安装好了Ubuntu就不用管驱动问题了,但是我们要用来搞学习的(Ubuntu自带显卡不行,TensorFlow必须要用英伟达的显卡+驱动),必须找到对1070最兼容的驱动。所以才有了下面这篇文章。

https://mp.csdn.net/postedit/79746624


先给大家讲一下我这次的各个版本吧:

系统:Ubuntu16.04

显卡:gtx170

显卡驱动:NVIDIA-Linux-x86_64-390.42.run===英伟达官网输入系统版本,显卡版本,自动匹配下载的

cuda:cuda_9.0.176_384.81_linux.run===因为tf1.6支持9.0,所以下载这个版本cuda

cudnn:cudnn-9.0-linux-x64-v7.tgz  ====cudnn给cuda加速的,自然版本要保持一致

anaconda:Anaconda3-4.4.0-Linux-x86_64.sh ===自己网上找的,好像不是最新版的。

TensorFlow:1.6.04.====最近出了1.7,貌似仅支持cuda9

https://www.tensorflow.org/serving/serving_basic


首先第一步肯定是装好Ubuntu,其次装好对应显卡驱动,现在我们按照上面的顺序开始安装cuda。


cuda安装:

安装CUDA9.0?当时官网说的最新的TensorFlow支持cuda9,但是最新的已经到了9.1.所以我装9.0版



1 。首先去官网下载CUDA 9.0,就截了一个图,下载的是.run文件。

百度自己推荐的,点进去就有了。

https://developer.nvidia.com/cuda-90-download-archive?

target_os=Linux&target_arch=x86_64&target_distro=Ubuntu&target_version=1604&target_type=runfilelocal


建议选择迅雷下载,360浏览器莫名其妙断了,下了一半的时候。迅雷也会断,但是你可以暂停一分钟,再去下快很多。也有人呢建议用火狐或者谷歌下,大家自己决定吧。


2. 找到Ubuntu你刚才下载文件里面安装


sudo sh cuda_9.0.176_384.81_linux.run

单击回车,一路往下运行,输入 accept 

直到提示“是否为NVIDIA安装驱动nvidia-384?”,选择   no,

后面问你是否安装 cuda toolkit 选yes

因为前面已经安装好驱动程序了,还要cuda自带的驱动干嘛,否则在这里你又要折腾一阵了。其他的全都是默认,不过要记住安装位置,它会问你输入路径,直接按 enter键 。默认是安装在/usr/local/cuda文件夹下。又问你是否安装在默认路径下,你输入yes。


3. 配置环境变量,运行如下命令打开profile文件
 sudo gedit  /etc/profile


打开文件后在文件末尾添加路径,也就是安装目录,命令如下:


export  PATH=/usr/local/cuda-9.0/bin:$PATH

export  LD_LIBRARY_PATH=/usr/local/cuda-9.0/lib64$LD_LIBRARY_PATH  


4. 保存,然后重启电脑
sudo reboot

5. 输入 

nvcc -V

有显示你的cuda信息就OK了,你也可以做下面的:


测试CUDA的例子
 cd  /usr/local/cuda-9.0/samples/1_Utilities/deviceQuery
  sudo make ./deviceQuery
如果显示的是关于GPU的信息,则说明安装成功了。



安装cudnn

1. 进入解压后的cudnn-9.0-linux-x64-v7.0.tgz文件,一般cudnn放在cuda安装包同一个文件夹里面
在终端执行下面的指令安装:

tar xzvf cudnn-9.0-linux-x64-v7.0.tgz
cd cuda    
sudo cp lib64/lib* /usr/local/cuda/lib64/    

sudo cp include/cudnn.h /usr/local/cuda/include/ 


2. 然后更新网络连接:
cd /usr/local/cuda/lib64/  

sudo chmod +r libcudnn.so.7.0.3  # cd /usr/local/cuda/lib64/ 目录下自己查看最新的 .so的版本,

                                 #因为还有一个老版本,这里是新的版本号  

sudo ln -sf libcudnn.so.7.0.3. libcudnn.so.7  
sudo ln -sf libcudnn.so.7 libcudnn.so  
sudo ldconfig  


anaconda安装


进入anaconda安装包目录下,

bash Anaconda3-4.4.0-Linux-x86_64.sh  一路默认即可,然后会一路安装各种依赖包。完成后,

最后一处,没有输入yes,报错了咋办?

Do you wish the installer to prepend the Anaconda3 install location
to PATH in your /root/.bashrc ? [yes|no]
[no] >>> 

You may wish to edit your .bashrc or prepend the Anaconda3 install location:

$ export PATH=/root/anaconda3/bin:$PATH

Thank you for installing Anaconda3!


Share your notebooks and packages on Anaconda Cloud!
Sign up for free: https://anaconda.org

方法:

export PATH=/root/anaconda3/bin:$PATH

最后别忘了重新加载一下。

source ~/.bashrc



anaconda环境下装TensorFlow

这里建议大家进tf官网学,同时参考我的这个教程。

https://tensorflow.google.cn/install/install_linux#InstallingAnaconda

https://tensorflow.google.cn/install/install_linux#the_url_of_the_tensorflow_python_package


1 .首先创建一个conda环境,命名为tensorflow

conda create -n tensorflow Python=2.7(如果是3.5版本就改成3.5)

现在终端输入 Python 就有显示版本信息了。


2.激活该环境,并在环境下安装tensorflow

source activate tensorflow
直接进入终端的tensorflow环境下:
这里写图片描述
接下来就可以安装tensorflow了,使用下命令:
pip3 install –ignore-installed –upgrade TF_PYTHON_URL(TF_PYTHON_URL是不同版
本的tensorflow源码的地址,我们这里安装的是Python 2.7的GPU版本)
pip3 install --ignore-installed --upgrade https://storage.googleapis.com/tensorflow/linux/gpu/tensorflow_gpu-1.6.0-cp36-cp36m-linux_x86_64.whl

 注意upgrade 与后面的链接之间有个空格。 这个链接是去官网找的最新版本的链接,日期不同可能不一样了。


如此便完成了GPU版本的tensorflow的安装。

3.使用完毕后,需要关闭tensorflow环境

source deactivate


4 。可以简单测试一下tensorflow安装是否成功:
这里建议大家一句一句的copy,不要一起都拷贝进去执行。

$ python

import tensorflow as tf
hello = tf.constant(‘Hello, TensorFlow!’)
sess = tf.Session() #在该步会显示电脑的显卡信息
print(sess.run(hello))
Hello, TensorFlow!
a = tf.constant(10)
b = tf.constant(32)
print(sess.run(a + b))

42

卸载tensorflow1.6

现在1.6 tf有bug,在终端上会无法使用tf,刚开始还是可以的,今天莫名其妙就挂了。在黑色命令窗口上却可以用。

刚开始我还卸载了,在home/conda3/envs/里面有前面安装好的tf文件,一句话就OK

sudo pip3 uninstall tensorflow

,然后又装了最新的tf1.7.



总结一下:

安装cuda,cudnn ,anaconda 基本无障碍的,很顺利的。开始安装tensorflow开始纠结哪种方法,因为有太多放方法了。

pip3 install --ignore-installed --upgrade https://storage.googleapis.com/tensorflow/linux/gpu/tensorflow_gpu-1.6.0-cp36-cp36m-linux_x86_64.whl

首先我的是Python3 ,所以要写pip3 ,所以后面安装到一半就要提示升级pip源,升级完了还一样,后来发现这里忘了官网给坑的

后面谷歌链接其实也挺快的,不用换到国内的源也行。几分钟也搞定。关键是官网的链接居然写的是CPU版,
还好官网下面也有gpu的链接,点进去copy出来。

整个过程是按照官网的教程来的,当然也看看国内的博客,是一样的。

最后有个巨坑!
如果是服务器装tf,OK,你再去装anaconda,装tf.建议pip法,有教程真的快捷。
https://blog.csdn.net/y1250056491/article/details/78670710

当然还有docker法,等其他方法度很麻烦。但稳妥点docker比较推荐,有点麻烦,

所以我选下面的anaconda法安装。


但是如果用自己的台式机,小心了,台式机Ubuntu自带Python,然后我们一般装anaconda(集成了大部分的Python库)所以省的自己搞。anaconda里面再pip法装tf。基本上成功,但是就是跑不起来tf,提示找不到不认识。
和我以前装caffe是一样的,按好了caffe就是找不到。
很好理解,你一个电脑里面有2Python,你让tf找哪个?肯定晕了。除非你知道怎么配置它。所以报错。

幸运的是 ,anaconda有专门的按装tf的方法,大家就不要偷懒用pip法了。用anaconda制定的方法保证OK。
我就是看官网做的,
https://tensorflow.google.cn/install/install_linux#InstallingAnaconda
https://tensorflow.google.cn/install/install_linux#the_url_of_the_tensorflow_python_package

https://blog.csdn.net/WIinter_FDd/article/details/66523468
https://blog.csdn.net/qq_18297933/article/details/52465616
第一次tf失败了,本以为要删除tf再来的,我没删除,直接再装居然也行啊。
谷歌工程师做出来 的产品就是稳啊。

这篇关于Ubuntu16.04+gtx1070显卡+安装TensorFlowGPU版的文章就介绍到这儿,希望我们推荐的文章对编程师们有所帮助!



http://www.chinasem.cn/article/355422

相关文章

如何在Mac上安装并配置JDK环境变量详细步骤

《如何在Mac上安装并配置JDK环境变量详细步骤》:本文主要介绍如何在Mac上安装并配置JDK环境变量详细步骤,包括下载JDK、安装JDK、配置环境变量、验证JDK配置以及可选地设置PowerSh... 目录步骤 1:下载JDK步骤 2:安装JDK步骤 3:配置环境变量1. 编辑~/.zshrc(对于zsh

如何在pycharm安装torch包

《如何在pycharm安装torch包》:本文主要介绍如何在pycharm安装torch包方式,具有很好的参考价值,希望对大家有所帮助,如有错误或未考虑完全的地方,望不吝赐教... 目录在pycharm安装torch包适http://www.chinasem.cn配于我电脑的指令为适用的torch包为总结在p

在PyCharm中安装PyTorch、torchvision和OpenCV详解

《在PyCharm中安装PyTorch、torchvision和OpenCV详解》:本文主要介绍在PyCharm中安装PyTorch、torchvision和OpenCV方式,具有很好的参考价值,... 目录PyCharm安装PyTorch、torchvision和OpenCV安装python安装PyTor

Python Transformer 库安装配置及使用方法

《PythonTransformer库安装配置及使用方法》HuggingFaceTransformers是自然语言处理(NLP)领域最流行的开源库之一,支持基于Transformer架构的预训练模... 目录python 中的 Transformer 库及使用方法一、库的概述二、安装与配置三、基础使用:Pi

如何解决mmcv无法安装或安装之后报错问题

《如何解决mmcv无法安装或安装之后报错问题》:本文主要介绍如何解决mmcv无法安装或安装之后报错问题,具有很好的参考价值,希望对大家有所帮助,如有错误或未考虑完全的地方,望不吝赐教... 目录mmcv无法安装或安装之后报错问题1.当我们运行YOwww.chinasem.cnLO时遇到2.找到下图所示这里3.

Python 安装和配置flask, flask_cors的图文教程

《Python安装和配置flask,flask_cors的图文教程》:本文主要介绍Python安装和配置flask,flask_cors的图文教程,本文通过图文并茂的形式给大家介绍的非常详细,... 目录一.python安装:二,配置环境变量,三:检查Python安装和环境变量,四:安装flask和flas

Win11安装PostgreSQL数据库的两种方式详细步骤

《Win11安装PostgreSQL数据库的两种方式详细步骤》PostgreSQL是备受业界青睐的关系型数据库,尤其是在地理空间和移动领域,:本文主要介绍Win11安装PostgreSQL数据库的... 目录一、exe文件安装 (推荐)下载安装包1. 选择操作系统2. 跳转到EDB(PostgreSQL 的

Linux系统中卸载与安装JDK的详细教程

《Linux系统中卸载与安装JDK的详细教程》本文详细介绍了如何在Linux系统中通过Xshell和Xftp工具连接与传输文件,然后进行JDK的安装与卸载,安装步骤包括连接Linux、传输JDK安装包... 目录1、卸载1.1 linux删除自带的JDK1.2 Linux上卸载自己安装的JDK2、安装2.1

Linux卸载自带jdk并安装新jdk版本的图文教程

《Linux卸载自带jdk并安装新jdk版本的图文教程》在Linux系统中,有时需要卸载预装的OpenJDK并安装特定版本的JDK,例如JDK1.8,所以本文给大家详细介绍了Linux卸载自带jdk并... 目录Ⅰ、卸载自带jdkⅡ、安装新版jdkⅠ、卸载自带jdk1、输入命令查看旧jdkrpm -qa

MySQL Workbench 安装教程(保姆级)

《MySQLWorkbench安装教程(保姆级)》MySQLWorkbench是一款强大的数据库设计和管理工具,本文主要介绍了MySQLWorkbench安装教程,文中通过图文介绍的非常详细,对大... 目录前言:详细步骤:一、检查安装的数据库版本二、在官网下载对应的mysql Workbench版本,要是