似乎在anaconda中找不到Cuda Toolkit 7.5 和cuDNN v4,所以需要去官网手动下载安装。
https://developer.nvidia.com/cudnn
https://developer.nvidia.com/cuda-downloads
1、Cuda的安装
下载完Cuda Toolkit 7.5后执行:
bash ./XX.run
输入exit跳过协议,不安装driver,路径如果是在服务器上不能使用默认的路径,需要自己设置自己的路径。
打开主目录下的 .bashrc文件,一般都是在:
/home/usename/
下面,如果看不到,通过命令:
ls -a
显示所有隐藏文件。此时,打开文件:
vim ./.bashrc
修改路径为:
export PATH="/home/username/cuda-7.5/bin:$PATH"
export LD_LIBRARY_PATH="/home/username/cuda-7.5/lib64/:$LD_LIBRARY_PATH"
终端再次运行:
source ~/.bashrc
通过nvcc --version命令产看是否安装成功。
2、Cudnn的安装
通过解压,拷贝等一系列操作就安装成功了
tar xvzf cudnn-7.0-linux-x64-v4.0-prod.tgz
sudo cp cudnn-7.0-linux-x64-v4/cudnn.h /home/username/cuda/include
sudo cp cudnn-7.0-linux-x64-v4/libcudnn/* /home/username/cuda/lib64
sudo chmod a+r /usr/local/cuda/include/cudnn.h /usr/local/cuda/lib64/……
3、tensorflow的安装
tensorflow的0.9.0版本似乎也是很早的了,anaconda大概是直接装不了了,不过可以在conda的环境里用pip安装。
# Ubuntu/Linux 64-bit, GPU enabled, Python 2.7
# Requires CUDA toolkit 7.5 and CuDNN v4.
(tensorflow)$ pip install --upgrade https://storage.googleapis.com/tensorflow/linux/gpu/tensorflow-0.9.0-cp27-none-linux_x86_64.whl
最后检查一下tensorflow安装成功了吗
$ 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
>>>
成功啦!!