tensorflow1.4.0 GPU安装_在anaconda子环境安装tensorflow1.4.0-gpu

tensorflow1.4.0 GPU安装

1. 创建虚拟环境
   打开Anaconda Prompt (Anaconda)
   输入conda create --name tensorflow python=3.6
   输入y确认
   可以输入conda env list查看是否创建成功
   激活环境activate tensorflow
   退出 deactivate*

2. 安装tensorflow gpu1.4版本
   先激活环境activate tensorflow
   

有的会提示升级pip版本 python -m pip install --upgrade pip
清华镜像安装tensorflow
pip install tensorflow-gpu==1.4.0 -i https://pypi.tuna.tsinghua.edu.cn/simple
输入pip list可以查看是否安装成功

3. 安装cuda8
   https://developer.nvidia.com/cuda-80-ga2-download-archive
   

注 其他版本cuda对照

4. 安装cuddn V6
   https://developer.nvidia.com/rdp/cudnn-archive
   可能会需要注册
   

解压后将duDNN的三个文件夹复制到下图路径

5. 配置环境变量
   电脑-属性-高级系统设置-环境变量-双击path
   将三个路径加入环境变量path中
   C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0\bin
   C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0\lib\x64
   C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0\include

6. 测试
   打开Anaconda Prompt (Anaconda)
   激活环境activate tensorflow
   输入python,输入以下代码

import tensorflow as tf 
hello = tf.constant('Hello, TensorFlow!') 
sess = tf.Session() 
print(sess.run(hello))
1
2
3
4

7. 开发环境
   打开开始中的这个程序
   
   把环境切换为tensorflow，安装spyder, jupyter notebook