tensorflow1.4.0 GPU安装
-
创建虚拟环境
打开Anaconda Prompt (Anaconda)
输入conda create --name tensorflow python=3.6
输入y确认
可以输入conda env list查看是否创建成功
激活环境activate tensorflow
退出 deactivate* -
安装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可以查看是否安装成功
- 安装cuda8
https://developer.nvidia.com/cuda-80-ga2-download-archive
注 其他版本cuda对照
- 安装cuddn V6
https://developer.nvidia.com/rdp/cudnn-archive
可能会需要注册
解压后将duDNN的三个文件夹复制到下图路径
-
配置环境变量
电脑-属性-高级系统设置-环境变量-双击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 -
测试
打开Anaconda Prompt (Anaconda)
激活环境activate tensorflow
输入python,输入以下代码
import tensorflow as tf
hello = tf.constant('Hello, TensorFlow!')
sess = tf.Session()
print(sess.run(hello))
- 开发环境
打开开始中的这个程序
把环境切换为tensorflow,安装spyder, jupyter notebook