Windows10系统tensorflow-gpu安装

安装前请确保自己的显卡支持gpu加速,支持加速的gpu型号可在下面的链接中查询。

https://www.geforce.com/hardware/technology/cuda/supported-gpus

一、安装anaconda

  1.   推荐从清华镜像站下载,速度快。
  2.   根据自己系统选择安装包,现(18.11.21)推荐Anaconda3-5.3.0及更高版本,因为Tensorflow暂不支持Python3.7。
  3.   安装时勾选Add Anaconda to my PATH environment variable

二、安装cuda及cudnn

  1. 下载安装cuda,https://developer.nvidia.com/cuda-downloads
  2. 下载cudnn,https://developer.nvidia.com/cudnn ,(下载cudnn需要注册账号,请根据cuda版本选择相对应cudnn)
  3. 解压cudnn,将解压出来的三个文件夹下面的文件移动到对应的cuda文件夹下(cuda默认安装路径为 C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0)    
  4. 将cuda的lib和include文件夹添加到环境变量

三、下载安装tensorflow-gpu

  打开命令行,执行下面命令。

pip install tensorflow-gpu -i https://pypi.tuna.tsinghua.edu.cn/simple

猜你喜欢

转载自www.cnblogs.com/clearlove7777777/p/9996918.html