您的位置:首页 > 其它

Windows10下安装GPU版本的Tensorflow

2017-09-10 18:27 555 查看
1、确保自己的显卡支持CUDA,安装CUDA8.0

2、下载cudnn6,解压后,修改环境变量,添加到路径,如下图:



3、安装Python3.5版本的anaconda3

4、命令行输入命令  pip
install --upgrade tensorflow-gpu(没有pip的自行百度安装的方法)

5、个人安装之前遇到的问题:

安装的TensorFlow版本为1.3,之前看网上的教程使用的是cudnn5,import tensorflow的时候一直报错,

后来改为cudnn6之后就好了,个人认为是因为TensorFlow版本不一样的问题。
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: