基于Tensorflow的gpu版本安装历来受人诟病,现有网上的版本可以帮助我们安装,但是不够详细和全面,往往需要结合看很多篇博客才能解决问题。笔者结合自己的安装经历,特在此留下一步到位的安装教程,希望对大家有帮助!
1.确认电脑是否有gpu,同时查看是否为支持cuda的gpu:
这里给出英伟达官网算力链接:CUDA GPUs | NVIDIA Developer
2.在安装好Anaconda或者Miniconda之后,安装CUDA Toolkit 和cudnn,链接如下:CUDA Toolkit Archive | NVIDIA Developer
https://developer.nvidia.com/rdp/cudnn-download ,这里的cudnn需要注册,但是不繁琐,按照步骤注册下载就行。
注意:network版和local版其实区别不大,如果下载好,会出现C:\Program Files\NVIDIA GPU Computing Toolkit这个路径。