Ubuntu安装Pyenv+Anaconda2+PyCharm+OpenAI-gym+Cuda+TensorFlow (二)
来源:互联网 发布:行情软件下载 编辑:程序博客网 时间:2024/06/01 17:37
四. 安装Cuda
该部分参考自TensorFlow Installation Guide
根据TensorFlow官网的指定,gpu版本对于cuda8.0和cnDNN5.1版本运行效果最好,并且需要gpu的计算性能在3.0以上:
- 校验gpu计算能力
首先验证下gpu的计算能力,在网址:
https://developer.nvidia.com/cuda-gpus
中找到自己的gpu版本型号,查看计算性能是否大于3.0。 安装cuda8.0
在官网下载Cuda ToolKit套件:安装步骤如上图官网所示,默认的安装路径为/usr/local/cuda。安装完成后,可以查看下是否存在这个文件夹,存在的话即安装成功。
安装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*配置环境变量
打开~/.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
待续..
个人理解,如有错误请指出
- Ubuntu安装Pyenv+Anaconda2+PyCharm+OpenAI-gym+Cuda+TensorFlow (二)
- Ubuntu安装Pyenv+Anaconda2+PyCharm+OpenAI-gym+Cuda+TensorFlow (一)
- Ubuntu安装Pyenv+Anaconda2+PyCharm+OpenAI-gym+Cuda+TensorFlow (三)
- Ubuntu安装Pyenv+Anaconda2+PyCharm+OpenAI-gym+Cuda+TensorFlow (四)
- 安装TensorFlow(Ubuntu+CUDA+Cudnn)
- OpenAI gym
- ubuntu安装(win10双系统)+cuda+cudnn+tensorflow安装
- ubuntu安装TensorFlow及编辑器pycharm
- ubuntu pyenv 配置安装
- Tensorflow安装教程(Linux+Anaconda2)
- tensorflow(2)---linux,anaconda2,pycharm中调用tensorflow
- windows + CUDA + Anaconda + tensorflow + PyCharm
- ubuntu16.04 利用pyenv安装anaconda3+pycharm
- pycharm下安装opencv3到anaconda2
- (二)Ubuntu下安装tensorflow
- Play with OpenAI Gym in Ubuntu 16.04: Hello World
- GTX 1080+Ubuntu 16.04+Cuda 8.0+Python 2.7+Anaconda2 下tensorflw 的安装
- Windows7+anaconda2+python3+PyCharm+TensorFlow 环境搭建(无GPU)
- 虚拟化与云计算期末复习
- 1.two sum个人刷题小结
- 缓存方面的原理问题
- SeekBar动态更新Thumb图片
- Reorder List
- Ubuntu安装Pyenv+Anaconda2+PyCharm+OpenAI-gym+Cuda+TensorFlow (二)
- android中string.xml中%1$s、%1$d等的用法
- 22.OpenGL--纹理组合器函数
- androidStudio给工程添加依赖模块
- PostgreSQL索引的4种类型
- Lazy Load 图片懒加载
- 工业视觉检测
- win下 cmd 查询本机开了哪些端口 并查看开启端口的软件
- 12.28JS学习笔记