前面文章介绍了anaconda+CUDA+pyCUDA+boost_python+vs2015环境配置,这里给出链接:https://blog.csdn.net/whatuwannado/article/details/82468086
一.前置工作
本文基于上文进行TensorFlow+CUDA+Anaconda+windows环境搭建。
python:3.6
vs:vs2015
cuda:CUDA Toolkit 9.0
CuDNN:cudnn-9.0-windows10-x64-v7.1
boost_python:boost_python 1_67
pyCUDA:pycuda-2018.1
TensorFlow:TensorFlow 1.10
二.安装CuDNN
(1)Cudnn官方下载地址:https://developer.nvidia.com/rdp/cudnn-archive需要注册,不过也蛮简单的。毕竟官方的才是值得信赖的嘛!
下载后解压,分别复制到已安装的cuda的对应目录中。
三.安装TensorFlow
本文通过anaconda安装的TensorFlow。
(1)管理员权限打开cmd,输入conda create -n tensorflow pip python=3.5,选择yes
(2)接着输入activate tensorflow;pip install --ignore-installed --upgrade tensorflow-gpu
结果:
(3)测试是否安装成功
在激活TensorFlow的环境下,输入import tensorflow as tf。不报错即安装成功
(4)接下来到自己的IDE中更改一下解释器即可
博主习惯使用spyder。进入anaconda如下,可见多了一个命名为“TensorFlow”的运行环境,切换至该环境,安装你所需要的包即可。
恭喜!环境搭建完毕!
这里给出一些参考链接:https://www.tensorflow.org/install/install_windows