ubuntu18.04如何安装tensorflow gpu

1.安装cuda和cudnn

第一个是安装驱动

根据需要安装的cuda版本，安装对应的驱动，比如cuda10.1所需要的最小显卡驱动版本为418，我之前安装的415版本的驱动就不可以使用，所以需要卸载驱动(如果卸载驱动之后出现一些很奇怪的界面，无法用boot-repair修复的话，还是重装系统吧，感觉这样比较快)

sudo apt-get remove --purge nvidia*

之后禁用nouveau驱动一定要安装下文所述，否则会出现无法登陆进入桌面的情况

如果出现无法登陆桌面的情况，尝试重装系统https://gummary.github.io/2018/11/04/install-cuda-ubuntu/ 

或者使用boot-repair进行修复（修复可能要进行两次，但也比重装系统快！）https://blog.csdn.net/qq413187589/article/details/43273011

接着安装cudn（一定要注意版本对应的显卡驱动版本）

https://blog.csdn.net/qq_33200967/article/details/80689543#CUDNN_205

https://blog.csdn.net/yjt1325/article/details/84641062

2.安装anconda2(博客的链接是安装anconda3，但由于我使用的代码是基于python2，所以这里我安装的仍然是anaconda2)

https://blog.csdn.net/u012318074/article/details/77074665

注意安装完anaconda之后应该在~/.bashrc中导入环境变量

export PATH=/home/tj816/anaconda3/bin:$PATH

 

3. 安装tensorflow

第一个比较好，第二个作为参考

https://blog.csdn.net/Angela_qin/article/details/80921688

https://blog.csdn.net/m0_37864814/article/details/82112029

4.注意每次使用tensorflow的时候，都需要source activate tf

测试tf是否安装成功

https://blog.csdn.net/jiang_z_q/article/details/73264561