前言
之前配过一次,但没有配好,o(╯□╰)o,所以这次又重新进行配置了一遍,现记录如下,希望和我情况相同的同学能够少点障碍。
环境及问题描述
环境
Windows10 64 位系统
GTX1060 6G
问题
需要在windows系统配置 TensorFlow 的GPU版本, 本方案应该可行,只要和我情况一样(别的系统没试过,不过把win10的部分换成win7应该也差不多能行吧),
本文用到的所有文件均已上传到百度网盘, 这里直接分享给大家,当然你也可以直接去官网下载对应版本。
链接: https://pan.baidu.com/s/1aHlZMN9r1uNriccIAqvwhA 提取码: xehh
一共有四个文件,请按需下载!
1>安装cuda9.2
直接点击运行cuda_9.2.148_win10.exe,默认安装即可。
2>安装cudnn9.2
把cudnn-9.2-windows10-x64-v7.1.zip解压
解压后, 将cuda文件夹中的bin,include,lib这三个文件夹直接复制到CUDA9.2的目录(C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.2)里。
3>配置cuda环境变量
在环境变量中加入如下几项:
4>安装Anaconda3
点击运行Anaconda3-2018.12-Windows-x86_64.exe,默认安装即可。
5>创建 新的python环境
以管理员身份打开命令提示行(这点很重要)
创建一个新的python3.6的环境
conda create -n tf_gpu_py36 pip python=3.6
激活这个环境
activate tf_gpu_py36
6>安装TensorFlow gpu版本
转到文件的下载目录后执行下面的命令
pip install tensorflow_gpu-1.8.0-cp36-cp36m-win_amd64.whl
7>测试安装是否成功
在命令行窗口输入python进入python命令行
python
然后输入如下代码:
import tensorflow as tf
hello = tf.constant('hello')
ses = tf.Session()
如果出现下面的图片就说明安装成功了:
如果出现无法加载动态库的错误,那很有可能是因为你配置的环境变量还没有生效,重新打开一次命令行窗口,然后激活刚刚创建的环境后再次运行上述代码即可!
总结
基本配置流程大致就是这样,配置过程很快也很顺利,基本没有遇到啥问题。但是不能排除你会遇到谜一样的问题,但是无论遇到什么问题都可以上网找,鲁迅不是有这么一句名言嘛(忘了是谁说的了,那就暂且是鲁迅说的吧):
你要相信,你所遇到的问题,网上百分之九十九会有人遇到过! ----鲁迅