ubuntu 16.0.4 配置 tensorflow-gpu 1.1.0
今天配了一下tensorflow的gpu版本，感觉有几个坑，网上的资料也比较乱，就想给大家留个比较完整安装和排坑的文档。
我的gpu是gtx 960m
1.安装驱动
没有安装驱动应该是选择x server的，应该也有一个NVIDIA的选项，我试过从这里更改，失败了，应该是源的网速不好，或是缺依赖。
网上有很多教程说什么禁用x server 什么的感觉很麻烦，而且不是很靠谱，直接用ppa这个方法比较简单，我选的版本是375，其他版本没有测试。
1 sudo add-apt-repository ppa:graphics-drivers/ppa 2 sudo apt-get update && sudo apt-get install nvidia-375
出现问题就试试 --fix-missing
安装后可以重启，也可以不重启，我没有重启，不过后来重启关机大概15分钟左右。
安装后，可以在这里查看
2.安装cuda
https://developer.nvidia.com/cuda-downloads
选择对应版本，.deb安装简单
Installation Instructions:
sudo dpkg -i cuda-repo-ubuntu1604-8-0-local-ga2_8.0.61-1_amd64.deb sudo apt-get update sudo apt-get install cuda
我安装的时候，什么都没有填，直接安装好了
3.安装cuDNN
https://developer.nvidia.com/rdp/cudnn-download
要注册账号
这里注意下，我一开始下载了v6版本，但tensorflow-gpu 1.1.0 版本需要v5版本。我下载的cuDNN v5 (May 27, 2016), for CUDA 8.0
下载后解压到安装路径，理论上哪里都行，我放在home下。
ALL PLATFORMS
Extract the cuDNN archive to a directory of your choice, referred to below as <installpath>.
Then follow the platform-specific instructions as follows.
LINUX
cd <installpath>
export LD_LIBRARY_PATH=`pwd`:$LD_LIBRARY_PATH
Add <installpath> to your build and link process by adding -I<installpath> to your compile
line and -L<installpath> -lcudnn to your link line.
这是官方安装教程。
也就是这个部分出现了最多的问题。
我也记不清楚具体步骤，只能说下我出现的几个错误
>>> import tensorflow
Traceback (most recent call last):File "<stdin>", line 1, in <module>File "/home/kai/anaconda2/lib/python2.7/site-packages/tensorflow/__init__.py", line 24, in <module>from tensorflow.python import *File "/home/kai/anaconda2/lib/python2.7/site-packages/tensorflow/python/__init__.py", line 51, in <module>from tensorflow.python import pywrap_tensorflowFile "/home/kai/anaconda2/lib/python2.7/site-packages/tensorflow/python/pywrap_tensorflow.py", line 52, in <module>raise ImportError(msg)
ImportError: Traceback (most recent call last):File "/home/kai/anaconda2/lib/python2.7/site-packages/tensorflow/python/pywrap_tensorflow.py", line 41, in <module>from tensorflow.python.pywrap_tensorflow_internal import *File "/home/kai/anaconda2/lib/python2.7/site-packages/tensorflow/python/pywrap_tensorflow_internal.py", line 28, in <module>_pywrap_tensorflow_internal = swig_import_helper()File "/home/kai/anaconda2/lib/python2.7/site-packages/tensorflow/python/pywrap_tensorflow_internal.py", line 24, in swig_import_helper_mod = imp.load_module('_pywrap_tensorflow_internal', fp, pathname, description)
ImportError: libcudnn.so.5: cannot open shared object file: No such file or directoryFailed to load the native TensorFlow runtime.首先是要把所有cudnn文件考到cuda中，而似乎这里只用到了libcudnn.so.5
sudo cp cuda/include/cudnn.h /usr/local/cuda/include
后面我是先把/home/kai/cuda/lib64下的所有文件cp过去，出现上面的错误。
sudo cp cuda/lib64/libcudnn_static.a /usr/local/cuda/lib64sudo ldconfig /usr/local/cuda/lib64
用这个命令确认一下
看来libcudnn.so.5必须是一个链接。
sudo ln -s /home/kai/cuda/lib64/libcudnn.so.5 libcudnn.so.5 #pwd = /usr/local/cuda/lib64
再试一下，成功
ps:我在前面改了libcudnn.so.5的权限，不过不清楚有没有作用，如果按上面过程出现问题，可以改下权限，移除链接，再重新创建链接。
sudo chmod u=rwx,g=rx,o=rx libcudnn.so.5 #pwd = 解压cudnn出来的目录
而且，这个方法我只是成功导入了tensorflow
4.安装tensorflow-gpu
pip install tensorflow-gpu
转载于:https://www.cnblogs.com/dk666/p/6761562.html
ubuntu 16.0.4 配置 tensorflow-gpu 1.1.0相关推荐
- Ubuntu 16.04 LTS下编译GPU版tensorflow
Ubuntu 16.04 LTS下编译GPU版tensorflow 机器学习与数学 · 2016-06-10 13:51 作者: 比特小组 机器学习与数学出品 机器学习必然涉及到代码,本小组选择sci ...
- 关闭防火墙linux 16.04,如何在Ubuntu 16.04上配置和设置防火墙
在本文中,我们将学习–如何在Ubuntu 16.04上配置和设置UFW(防火墙),UFW代表"不复杂的防火墙",它充当IPTABLES的接口,从而简化了防火墙的配置过程,对于防火墙 ...
- tesla p4 linux驱动,Ubuntu 16.04. 装tesla p4 显卡驱动+cuda9.0+docker+nvidia-docker 详细方法，这里是服务器为主...
Ubuntu 16.04. 装tesla p4 显卡驱动+cuda9.0+docker+nvidia-docker 详细方法,这里是服务器为主 这里 说明一下,我也是在网上看的教程,小白一个,通过好几 ...
- Ubuntu 16.04下配置openWRT开发环境
Ubuntu 16.04下配置openWRT开发环境 1.配置openwrt开发环境,本实验室在ubuntu16.04系统下测试 在编译之前我们先要搭建环境,ubuntu下OpenWrt编译环境需要安 ...
- Ubuntu 16.04.4 配置 Nvidia显卡驱动 + CUDA 9.0 + cuDNN v7.0.5 + OpenCV 3.2.0
1. 安装Nvidia显卡驱动 1.1 卸载原驱动 .run文件卸载 sh NVIDIA-Linux-x86_64-390.77.run --uninstall apt-get 卸载 apt-get ...
- 来自官方文档的Ubuntu 16.04 + tensorflow-GPU 配置
I Preprare for CUDA installation 官方文档:http://docs.nvidia.com/cuda/cuda-installation-guide-linux/ind ...
- 玩玩机器学习1——ubuntu16.04 64位安装TensorFlow GPU+python3+cuda8.0+cudnn8.0
如今的机器学习运算大多数都是利用gpu进行,包括很大大型游戏,比特币的开采,都离不开GPU的运算,GPU已成为大型运算能力的主要硬件. 大名鼎鼎的机器学习开源框架,如TensorFlow和Caffe, ...
- 在ubuntu 16.04上安装tensorflow，并测试成功
用下面代码测试安装: 1 #! /usr/bin/python 2 # -*- coding: utf-8 -*- 3 4 import tensorflow as tf 5 import numpy ...
- linux创建django项目,Ubuntu 16.04下配置Django项目
记录一次在全新服务器下配置Django项目说明: Linux操作系统:Ubuntu 16.04 Django版本:1.8.7 Python版本:2.7.12 运行模式:Ubuntu+Django+Ap ...
- peazip 64 linux,Ubuntu 16.04中下载安装PeaZip 6.5.0档案管理器
PeaZip是一款免费开源的图形化档案管理器和压缩工具,它可以安装在Linux.BSD.Windows操作系统上,目前最新版本是PeaZip 6.5.0.它支持180种以上档案文件格式,包括rar.z ...
最新文章
- linux 路由跟踪表满错误 nf_conntrack: table full, dropping packet 原理解决方法
- zeekooper集群搭建_Zookeeper与Kafka集群搭建完整教程
- java 后台输出_将后台输出动态打输出到前台并且显示
- 创建自已的sql函数
- [UE4]报错：Material with missing usage flag was applied to skeletal mesh 的解决方法：为材质设置相应的 usage flag
- 删除按钮_系统小技巧：一站式完成系统“删除”性维护
- silverlight 自定义资源整理（待后续补充）
- Python输出当前时间
- 写代码时发现......还是SpringBoot牛逼！
- 使用Pack200压缩你的代码
- 【LeetCode】59. Spiral Matrix II
- Socket编程--TCP/UDP
- Linux常用命令大全 阶段性总结（二）
- java 文字串叠字检查_正则表达式的相关应用
- 哔哩哔哩笔试 01串求最长1子串的长度
- swift 调用震动
- JVM 工作原理和即时编译（JIT）
- 点评“最好的300款免费软件”
- MINI PCI-E 引脚定义
- 机器学习数据集篇——KMNIST数据集