您的位置:首页 > 其它

Windows下TensorFlow-gpu版安装教程

2017-11-05 17:07 441 查看
1.安装cuda8,https://developer.nvidia.com/cuda-downloads

安装完在cmd里输入nvcc -V验证是否安装成功

2.下载cudnn6,是cudnn6!

3.解压cudnn6,将里面的三个文件夹拷到C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\中

网上很多教程都漏了最后一步:环境变量的添加

在系统变量Path中加入:C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0\bin;

4.至此GPU加速就配置好了,接下来在cmd里输入pip install tensorflow-gpu就行了
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: