ubuntu16安装tensorflow gpu加速(包括cuda8及NVIDIA显卡安装)

阅读: 评论:0

ubuntu16安装tensorflow gpu加速(包括cuda8及NVIDIA显卡安装)

ubuntu16安装tensorflow gpu加速(包括cuda8及NVIDIA显卡安装)

1. 安装tensorflow gpu版本 :

pip install tensorflow-gpu

如如果安装cpu版本的,可以用:pip install tensorflow

如果是python3及以上,需要先安装pip3

sudo apt-get install python3-pip
以后可能需要的几个依赖包在这里就一起安装了:
sudo apt-get install build-essential libssl-dev libffi-dev python3-dev
附,python3.5安装(ubuntu可以python2.7和3.5共存,但是只能用一个):

(1)sudo add-apt-repository ppa:fkrull/deadsnakes

(2)sudo apt-get update

(3)sudo apt-get install python3.5

(4)sudo cp /usr/bin/python /usr/bin/python_bak,先备份 (5)sudo rm /usr/bin/python 删除 (6)sudo ln -s /usr/bin/python3.5 /usr/bin/python

默认设置成python3.5,重建软链接这样在终端中输入python默认就是 3.5版本了

2. 安装显卡驱动,


(1)接下来开始正式安装显卡驱动

2.1、禁用系统默认的集成驱动

Ubuntu系统集成的显卡驱动程序是nouveau,它是第三方为NVIDIA开发的开源驱动,我们需要先将其屏蔽才能安装NVIDIA官方驱动。
将驱动添加到黑名单f中,但是由于该文件的属性不允许修改。所以需要先修改文件属性。

查看属性
   sudo ls -lh /etc/modprobe.f

修改属性
    sudo chmod 666 /etc/modprobe.f

用gedit打开
    sudo gedit /etc/modprobe.f

在该文件最后后添加:


blacklist nouveau

2.2.开始安装显卡

 2.2.1打开系统设置


点击软件和更新,选择附加驱动,选择上面的显卡驱动,然后点击应用更改即可,完了之后重启



3.安装cuda8以及cudnn

3.1 :cuda8下载地址:

cudnn下载地址:     (需注册才能下载)

也可以到百度云盘一起下载,省去注册麻烦: 

下载好之后接下来正式开始安装,

3.2: 先解决相关依赖安装:

sudo apt-get install freeglut3-dev build-essential libx11-dev libxmu-dev libxi-dev libgl1-mesa-glx 
libglu1-mesa libglu1-mesa-dev 如果安装依赖过程中最后出现了(没有出现直接无视跳到下面安装cuda8进行安装):

/al: /usr/lib/nvidia-375/libEGL.so.1 is not a symbolic link
/al: /usr/lib32/nvidia-375/libEGL.so.1 is not a symbolic link
不是连接符号,

解决方法是:

sudo mv /usr/lib/nvidia-375/libEGL.so.1 /usr/lib/nvidia-375/libEGL.so.1 sudo mv /usr/lib32/nvidia-375/libEGL.so.1 /usr/lib32/nvidia-375/libEGL.so.1 sudo ln -s /usr/lib/nvidia-375/libEGL.so.375.39 /usr/lib/nvidia-375/libEGL.so.1 sudo ln -s /usr/lib32/nvidia-375/libEGL.so.375.39 /usr/lib32/nvidia-375/libEGL.so.1


3.3 安装cuda8
进入下载目录,输入命令:
    sudo sh cuda_8.0.61_375.26_linux.run --override
到了这个界面之后按住:Ctrl+c跳过

接下来,输入accept
下一步,提示安装显卡驱动,输入n,要不然会出错


接下来全部yes,有选择安装路径,的,直接默认就好,不用输入,下面是所有的输入
,然后可以去喝杯茶等几分钟就可以安装好了。


出现了这个,说明安装完成:
接下来导入cuda8的配置路径,如果是按默认路径安装的,直接输入以下命令:

export PATH="$PATH:/usr/local/cuda-8.0/bin"
export LD_LIBRARY_PATH="/usr/local/cuda-8.0/lib64"

输入:nvidia-smi,显示下面信息则说明正确安装了



3.4 然后开始安装深度学习库cudnn

3.4.1.解压刚刚下载的cudnn :
tar xvzf cudnn-8.
2.然后将库和头文件copy到cuda目录(一定是你自己安装的目录如/usr/local/cuda-8.0)
执行下面命令复制文件(请注意自己的解压路径)
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*


cudnn复制完成就行,不用安装

cudnn复制完成之后,再输入(解决可能会出现的连接符错误):

sudo cp /usr/local/cuda-8.0/lib64/libcudart.so.8.0 /usr/local/lib/libcudart.so.8.0 && sudo ldconfig
sudo cp /usr/local/cuda-8.0/lib64/libcufft.so.8.0 /usr/local/lib/libcufft.so.8.0 && sudo ldconfig
sudo cp /usr/local/cuda-8.0/lib64/libcurand.so.8.0 /usr/local/lib/libcurand.so.8.0 && sudo ldconfig
sudo cp /usr/local/cuda-8.0/lib64/libcublas.so.8.0 /usr/local/lib/libcublas.so.8.0 && sudo ldconfig
sudo cp /usr/local/cuda-8.0/lib64/libcudnn.so /usr/local/lib/libcudnn.so && sudo ldconfig
sudo cp /usr/local/cuda-8.0/lib64/libcudnn.so.5 /usr/local/lib/libcudnn.so.5 && sudo ldconfig
sudo cp /usr/local/cuda-8.0/lib64/libcudnn.so.5.1.10 /usr/local/lib/libcudnn.so.5.1.10 && sudo ldconfig
(ps,如果不是在百度云盘下载的cudnn,链接最后一条的时候需要到
/usr/local/cuda-8.0/lib64/下查看是哪个文件,不一定是5.1.10,可能是
libcudnn.5.1.15或者其他的)最后,检测tensorflow,
在终端输入:python
然后进入之后输入:import tensorflow,如果没有报错则说明已经配置好了

引用:
opencv安装:.html

3、测试opencv是否安装成功

1 root@VB:# python
2 >>>import cv2  
3 >>>cv2.__version__
4 '3.2.0'

  显示出版本号,则表示opencv安装成功。

 

《注意》:如果打开python后,import cv2时,提示:

1 >>> import cv2
2 Traceback (most recent call last):
3   File "<stdin>", line 1, in <module>
4 ImportError: No module named cv2

则需要安装python-opencv模块,在命令窗口中键入: pip install opencv-python

安装成功后,即可导入cv2,此时能够正确导入cv2,且正确显示版本号:

1 >>> import cv2
2 >>> cv2.__version__
3 '3.2.0'


本文发布于:2024-02-01 04:22:16,感谢您对本站的认可!

本文链接:https://www.4u4v.net/it/170673253733822.html

版权声明:本站内容均来自互联网,仅供演示用,请勿用于商业和其他非法用途。如果侵犯了您的权益请与我们联系,我们将在24小时内删除。

标签:显卡   tensorflow   gpu   NVIDIA
留言与评论(共有 0 条评论)
   
验证码:

Copyright ©2019-2022 Comsenz Inc.Powered by ©

网站地图1 网站地图2 网站地图3 网站地图4 网站地图5 网站地图6 网站地图7 网站地图8 网站地图9 网站地图10 网站地图11 网站地图12 网站地图13 网站地图14 网站地图15 网站地图16 网站地图17 网站地图18 网站地图19 网站地图20 网站地图21 网站地图22/a> 网站地图23