Windows下TensorFlow-gpu版安装教程

来源:互联网 发布:linux 批量删除进程 编辑:程序博客网 时间:2024/06/05 19:27

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就行了


原创粉丝点击