机器学习环境搭建:GTX970+Ubutnu1404_64bit+TensorFlow(GPU)
来源:互联网 发布:md5算法java实现 编辑:程序博客网 时间:2024/04/29 22:51
TF的GPU版本需要安装CUDA和cuDNN,而在Ubuntu中安装显卡驱动稍微麻烦一点。下面的安装比较简单:
1.检查显卡支持情况
https://developer.nvidia.com/cuda-gpus
2.下载一些东西
显卡的驱动:http://www.geforce.cn/drivers
cuda7.5 tool kit(需要注册):https://developer.nvidia.com/cuda-downloads(下载runfile就可以了)
cuDNN(需要注册):https://developer.nvidia.com/rdp/cudnn-download(简单的选择v4,v5需要TF从源码安装)
(图片是对应的下载界面)
3.安装显卡驱动
简要说明:Ubuntu的GPU驱动安装稍微麻烦一点,而安装tool kit会提示是否安装显卡驱动,问题是顺着tool kit安装显卡驱动是走不通的,所以显卡驱动要单独安装。
麻烦点:只要N卡运行,X server服务就有,需要手动关闭:
ctrl+alt+F1 进入命令行模式,关掉方式:sudo /etc/init.d/lightdm stop,然后开始安装夏卡驱动:
直接运雄驱动的*.run文件即可。中间会有提示,选yes没有害处,可能会提示卸载之前的驱动,选yes即可。最后提示安装成功,会回到图像桌面,手动重启一下。
4.安装tool kit
同样进入命令行,关掉X server服务,运行run文件。除了驱动不安装外其他都是yes。安装完手动重启一下。
5.安装cuDNN
这个是把一些文件复制到指定位置即可:
tar xvzf cudnn-7.0-linux-x64-v4-prod.tgzsudo cp cuda/include/cudnn.h /usr/local/cuda/includesudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64sudo chmod a+r /usr/local/cuda/include/cudnn.h /usr/local/cuda/lib64/libcudnn*
6.添加环境变量否则TF不知道CUDA的东西在哪里,方法:在/etc/environment文件中直接添加下面路径:
:/usr/local/cuda/lib64:/usr/local/cuda
7.安装pip和TF
默认安装TF对应使用Python2.7,想用其他的可以参考官网:https://www.tensorflow.org/versions/r0.10/get_started/os_setup.html#pip-installation
sudo apt-get install python-pip python-dev
export TF_BINARY_URL=https://storage.googleapis.com/tensorflow/linux/gpu/tensorflow-0.10.0rc0-cp27-none-linux_x86_64.whl
sudo pip install --upgrade $TF_BINARY_URL
8.测试
进入Python测试:(显示一些输入输出)
$ 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>>>
9.可能的错误
ImportError: libcudart.so.7.0: cannot open shared object file: No such file or directory
是环境变量没设置好
0 0
- 机器学习环境搭建:GTX970+Ubutnu1404_64bit+TensorFlow(GPU)
- Ubuntu16.04下GTX970配置机器学习环境
- TensorFlow(GPU支持)环境搭建
- win10 tensorflow-gpu 环境搭建
- 使用Amazon AWS搭建GPU版tensorflow深度学习环境
- 深度学习平台TensorFlow环境搭建【GPU版】
- Windows下TensorFlow环境搭建(无GPU)
- TensorFlow GPU环境linux搭建指南
- 【Setting】win10+python3+tensorflow-gpu环境搭建
- Centos6.3 keras+tensorflow+xgboost+libsvm机器学习环境搭建
- TensorFlow笔记2:Ubuntu系统+Tensorflow-gpu环境搭建
- 配置tensorflow GPU深度学习环境
- 深度学习GPU环境搭建:ubuntu16.04+GTX1070+Cuda8.0+tensorflow build from source
- 深度学习tensorflow-gpu环境搭建避坑指南-win10_anaconda_python3.5_cuda8.0
- Tensorflow C++ 学习(一) 搭建环境
- Tensorflow C++ 学习(一) 搭建环境
- 机器学习环境搭建
- 机器学习环境搭建
- Green VPN 优惠码
- HDU 2488 A Knight's Journey DFS
- 输入输出流>>和<<操作符对字符串操作的一些问题详解
- Android抛出异常NetworkOnMainThreadException解决方案
- mysql 汇总数据 学习笔记
- 机器学习环境搭建:GTX970+Ubutnu1404_64bit+TensorFlow(GPU)
- POJ-1611 并查集-The Suspects
- 微信H5支付开发步骤
- iOS 知识小集(横竖屏切换)
- 判断树T2是否为T1的子树
- JQuery初识
- 装载问题——搜索回溯算法
- C++编程好习惯
- 微信H5支付开发步骤