前言
今天打算在新机子上安装TensorFlow-GPU版本,网上搜教程,发现都是一些古老教程,很多都是说不支持python3.7,需要新建一个3.6环境再去安装。为了防止纯小白采坑,特意写一篇超简单教程,也给自己备忘一下。
我的环境是:anaconda python3.7+cuda10.1+cudnn7.6
想直接安装的请直接看:正规安装流程
采坑部分
于是我去python第三方包搜索网站搜了一下:https://pypi.org/project/tensorflow-gpu/#files。TensorFlow已经出到了最新2.1.0版本完全支持python3.7版本。
我这个人又很懒不想下载包再去安装,所以搜了一下TensorFlow2的手册,于是发现官网说:https://tensorflow.google.cn/install
#更新pip
pip install --upgrade pip
#安装tensorflow
pip install tensorflow
好吧,就是这么简单?
为了防止版本有什么问题,我写的:
pip install tensorflow-gpu==2.1.0
当时觉得自己太牛了,装完之后一测试。
我眼前的这都是啥,满屏幕都装不下的error。心都碎了,于是
我就把anaconda全部卸掉,重新来过。。。。
正规安装流程
tensorflow2.1.0需要英伟达的支持,还需要安装cuda和cudnn
cuda对应的版本是:10.1,cudnn对应的版本是:7.6
anaconda官网下载无脑安装就可以了。
接着官网下载cuda: https://developer.nvidia.com/cuda-toolkit-archive
或者:conda install cudatoolkit=10.1
cudnn: https://developer.nvidia.com/rdp/cudnn-archive
或者:conda install cudnn=7.6
最后安装tensorflow-gpu2.1.0:
pip install tensorflow-gpu==2.1.0
到现在为止全部安装完成,绝大部分胖友的tensorflow都可以使用了,但是!
有的胖友会出现可怕的报错:importError:DLL load failed:找不到指定的模块。
这个问题折腾了我一天,最后亲测有效方式如下:
下载并安装:适用于 Visual Studio 2015、2017 和 2019 的 Microsoft Visual C++ 可再发行软件包
好了各位胖友,到此为止,我踩的坑已经全部解决,希望可以帮助到大家。