详细安装Tensorflow gpu版本

基于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这个路径。

当然上述安装并不是一定要最新版,按照个人需要就行。 同时官网也给出了已经测试稳定的推荐搭配,详见:从源代码构建  |  TensorFlow

 根据表格中的搭配,选择下载对应的CUDA和cudnn。

 

3.下面就可以进入命令行进行环境的安装操作了;笔者给出成功率很高的安装推荐:Tensorflow-gpu安装超详细!!!_东海扬尘_BingO的博客-CSDN博客_tensorflowgpu安装

按照步骤可以完成gpu的安装,安装过程中,如果出现不懂的,可以在评论中提出! 

你可能感兴趣的:(tensorflow)