windows10 安装tensorflow—GPU版

步骤:首先安装CUDA,然后下载对应版本的cuDNN,最后安装tensorflow(pip 即可)

注意:显卡是否支持CUDA(NVIDIA显卡一般都支持),显卡驱动的版本与CUDA版本的对应表(https://docs.nvidia.com/cuda/cuda-toolkit-release-notes/index.html),CUDA版本与对应版本的cuDNN。

详情参考:

  • https://blog.csdn.net/vvvvs13/article/details/106262303/

  • https://blog.csdn.net/weixin_46161549/article/details/116700835

原文地址:https://www.cnblogs.com/-hao-/p/15468175.html