Tensorflow gpu win10
1.安装cuda 9.0
2.解压cudnn-9.0-windows10-x64-v7.1.zip
复制解压文件目录下 bin,include,lib 三个文件夹到 cuda根目录
(默认在C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0)
3.安装Anaconda 3
4.打开Anaconda Prompt,创建环境
conda create -n tf-gpu python=3.6
5.进入环境
conda activate tf-gpu
6.升级pip到最新版,防止稍后的安装时,出现错误
python -m pip install --upgrade pip
7.安装tensorflow-gpu版本 (注:奔腾系列处理器可能需要低版本,tensorflow==1.5,原因:不支持AVX指令集)
pip install tensorflow-gpu ==1.11.0
8.验证:在命令行中,进入python,并输入以下代码:
import tensorflow as tf
hello = tf.constant('Hello, TensorFlow!')
sess = tf.Session()
print(sess.run(hello))
输出如下,即为环境配置成功
b’ Hello, TensorFlow
注:1.如果import tensorflow时出现以下的错误,可以检查下处理器是否为奔腾系列,是否支持AVX指令集
换成低版本的tensorflow即可 (如1.5)
ImportError: DLL load failed: 动态链接库(DLL)初始化例程失败。