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

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值