Ubuntu安装Pyenv+Anaconda2+PyCharm+OpenAI-gym+Cuda+TensorFlow （二）

四. 安装Cuda

　　该部分参考自TensorFlow Installation Guide
　　根据TensorFlow官网的指定，gpu版本对于cuda8.0和cnDNN5.1版本运行效果最好，并且需要gpu的计算性能在3.0以上：

1. 校验gpu计算能力
   　　首先验证下gpu的计算能力，在网址：
   　　https://developer.nvidia.com/cuda-gpus
   　　中找到自己的gpu版本型号，查看计算性能是否大于3.0。

2. 安装cuda8.0
   　　在官网下载Cuda ToolKit套件：
   

   

   　　安装步骤如上图官网所示，默认的安装路径为/usr/local/cuda。安装完成后，可以查看下是否存在这个文件夹，存在的话即安装成功。

3. 安装cuDNN5.1
   　　在网址：
   　　https://developer.nvidia.com/cudnn
   　　下载cuDNN5.1，（需要先注册一个账号）
   

   

   　　然后将它放在我们cuda toolkit安装的地方，也就是/usr/local/cuda文件夹下。在bash中进到这个文件夹中，运行如下命令：

   tar xvzf cudnn-8.0-linux-x64-v5.1-ga.tgz
   sudo 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*

4. 配置环境变量
   　　打开~/.bashrc文件,添加如下环境变量：

   export LD_LIBRARY_PATH=”$LD_LIBRARY_PATH:/usr/local/cuda/lib64:/usr/local/cuda/extras/CUPTI/lib64”
   export CUDA_HOME=/usr/local/cuda

五. 安装TensorFlow

　　该部分主要参考TensorFlow官网中Anaconda installation部分。主要安装步骤如下：

　　在第三节中利用pyenv安装了anaconda，因此这里我们可以直接从第二步开始一直进行到第四步。第五步表明我们可以继续安装其它的python package，比如说最后我们要安装的openai-gym。
　　
1. 首先创建一个叫tensorflow的虚拟环境：

# Python 2.7
$ conda create -n tensorflow anaconda

　　注意这里我们创建这个环境用了anaconda版本的python，与官网上给出的python=2.7不同，这样我们才可以把tensorflow整合到anaconda中。
　　
2. 然后，激活这个环境：

$ source activate tensorflow
(tensorflow)$ # Your prompt should change

　　需要注意的是，这一步一开始是会出错的。在bash中输入source activate tensorflow会提示pyenv：bash：commond not found，然后整个bash就自动关闭了。这是因为我们的anaconda是利用pyenv安装的，bash从当前的环境变量暂时还找不到这个anaconda；而官网则是从anaconda的安装包直接安装到本地的，在安装的最后还会自动帮我们设定好它的环境变量，所以这里我们还需要自己把pyenv环境下anaconda的环境变量添加到~/.bashrc文件。
　　打开~/.bashrc文件,在pyenv init后添加如下环境变量：

export PATH=”$PYENV_ROOT/versions/anaconda2-4.2.0/bin:$PATH”

　　设置好之后，再激活tensorflow环境，这下就能成功进入啦：

3.设定tensorflow的安装地址：

# 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.
(tensorflow)$ export TF_BINARY_URL=https://storage.googleapis.com/tensorflow/linux/gpu/tensorflow_gpu-0.12.0-cp27-none-linux_x86_64.whl

4.安装：

# Python 2
(tensorflow)$ pip install –ignore-installed –upgrade $TF_BINARY_URL

至此，TensorFlow就被安装到pyenv所维护的anaconda2-4.2.0版本下的tensorflow环境中了。

全文参考：
[1]. 南墙已破的简书http://www.jianshu.com/p/c89b97d052b7
[2]. TensorFlow官网https://www.tensorflow.org/get_started/os_setup#anaconda_installation
[3]. OpenAI-gym官网https://gym.openai.com/docs
[4]. 知乎专栏-智能单元https://zhuanlan.zhihu.com/intelligentunit
[5]. super的博客园http://www.cnblogs.com/super-d2/p/4725818.html

待续..
个人理解，如有错误请指出

　　　　　　　

（转载请注明作者和出处： http://blog.csdn.net/gh234505 未经允许请勿用于商业用途）