GTX1080+ubuntu14.04+cuda8.0+cudnn5.1+tensorflow

来源:互联网 发布:大学生程序员 编辑:程序博客网 时间:2024/05/01 21:42

吐嘈:

作为一个资深的linux小白,当初给我实验室电脑装双系统都折腾了半天。

安装ubuntu14.04的时候,出现了很多问题。最终用VGA信号才正常了。这个容后再说。


装cuda

cuda8.0官网下载

我下载的是.deb版本的

按照官网给的命令

cd到下载的.deb文件的目录

  1. `sudo dpkg -i cuda-repo-ubuntu1404-8-0-local_8.0.44-1_amd64.deb`
  2. `sudo apt-get update`
  3. `sudo apt-get install cuda
之后设置环境变量

sudo gedit /etc/profile

在文件的最后加上

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

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

并保存

然后,一定要!!

source /etc/profile

使环境变量生效


之后,在终端输入 nvcc -V

看一下cuda的版本是不是8.0


安装 cudnn5.1

cudnn安装

需要注册个nvidia的帐号,填两个调查问卷

下载

cudnn-8.0-linux-x64-v5.1.tgz

然后

tar xvzf cudnn-8.0-linux-x64-v5.1.tgzsudo cp -P cuda/include/cudnn.h /usr/local/cuda/include/sudo cp -P cuda/lib64/libcudnn* /usr/local/cuda/lib64/sudo chmod a+r /usr/local/cuda/include/cudnn.h /usr/local/cuda/lib64/libcudnn*
然后

$ sudo apt-get install libcupti-dev

应该就没问题了。


安装tensorflow

极客学院 tensorflow的中文文档

pip GPU安装,只更新到

 Ubuntu/Linux 64-bit, GPU enabled, Python 2.7. Requires CUDA toolkit 7.5 and CuDNN v4.

安装cuda8.0和cudnn5.1GPU版本的tensorflow需要到tensorflow英文官网,或者下面这个网站

tensorflow安装教程

我的python是2.7版本的

先装pip

# Ubuntu/Linux 64-bit$ sudo apt-get install python-pip python-dev

然后GPU版本的tensorflow

【高亮注意】这里给的tensorflow版本是1.0.0-rc2的!与0.12.0有一些变化变化

一些指令(函数?)变了,就会出现AttributeError: 'module' object has no attribute 'XXXXXX'

然后我想卸了1.0.0的安旧版的,结果还卸不掉,只能先再安个旧版的试试,实在不行就只能重装系统了。。。。。

# Ubuntu/Linux 64-bit, GPU enabled, Python 2.7# Requires CUDA toolkit 8.0 and CuDNN v5. For other versions, see "Installing from sources" below. $ export TF_BINARY_URL=https://storage.googleapis.com/tensorflow/linux/gpu/tensorflow_gpu-1.0.0rc2-cp27-none-linux_x86_64.whl
安装tensorflow

# Python 2$ sudo pip install --upgrade $TF_BINARY_URL

这就安装完了,可以测试一下

终端输入:

$ 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>>>


cuda简直太坑了!!!!!

tensorflow的版本也很坑啊!!!!!!

0 0
原创粉丝点击