Win10配置tensorflow_gpu真实踩坑记录

周末闲着无事就打算玩玩unity的machine learning,可能就着官网(https://github.com/Unity-Technologies/ml-agents/blob/master/docs/Installation-Windows.md)这里一样配置就不会出这么多问题,但是自己作,要不下载更新的版本或者旧的版本。

配置

Vs2017

Cuda_9.0.176_win10

Cudnn9.0

Anaconda 4.5.11

Python3.6

开始很顺利,直到配置tensorflow_gpu各种各样问题出现了。

安装tensorflow1.12.0

1.打开Anaconda Prompt,进入Anaconda命令行管理界面。配置清华仓库镜,输入指令:

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/

国内可能由于网络差,会出现下载各种timeout,网上各种大神说多试几遍,反正在我这都没有效果,配置镜像我还是一样,最后我还是(VPN)下载好了。

2.创建运行环境,输入指令:

conda create -n tensorflow-gpu python=3.6

新建一个名字叫“tensorflow-gpu”,python版本为3.6的运行环境,此环境与Anaconda中其它环境隔离。软件包也会随之安装,输入“y“和回车后开始安装。

3.激活并进入环境,使后续指令在激活的环境中生效,输入指令:

conda activate tensorflow-gpu

4.升级pip到最新版,防止稍后的安装时,出现错误,输入指令:
python -m pip install --upgrade pip

5.安装tensorflow1.12.0及相应依赖包,输入指令:
pip install --ignore-installed --upgrade tensorflow-gpu

至此,tensorflow的安装完成。

验证

在命令行中,进入python,并输入以下代码:

import tensorflow as tf

hello = tf.constant('Hello, TensorFlow!')

sess = tf.Session()

print(sess.run(hello))

输出:
b’ Hello, TensorFlow!

在最后验证时候,在此我卡了很久,一直提示Dll load failed 找不到指定的模块。各种尝试,起先怀疑python版本太新了,后面尝试安装cpu版本的tensorflow是OK,所以定位到cuda安装和cudnn应该是版本不匹配。最早先安装时Cuda8,不行,后面直接安装Cuda9.2,仍是不行,继续尝试Cuda9.0和Cudnn9,大功告成。

当Hello,TensorFlow出现时候,都快内流满面了。

你可能感兴趣的:(Tensorflow)