Windows10 安装TensorFlow-gpu 1.9.0

保证‘Windows Installer’服务是开启状态

右键计算机>管理>服务和应用程序>服务
找到“Windows Installer”
右键选择“启动”

安装CUDA与CUDNN

根据现最新版本TensorFlow1.9.0,其所要求的cuda版本号为9.0,cudnn版本号为7.0.5。

  • CUDA 9.0 https://developer.nvidia.com/cuda-toolkit-archive
  • cuDNN 7.0.5 https://developer.nvidia.com/rdp/cudnn-archive
    按照要求安装即可,最后需要将cudnn添加进环境变量D:\cudnn\bin

安装tensorflow-gpu1.9.0

pip install tensorflow-gpu

你可能感兴趣的:(Windows10 安装TensorFlow-gpu 1.9.0)