前言
电脑配置Deepin 20 + GTX1050,tensorflow-gpu安装的版本是tensorflow-gpu 2.0.0,Python3.7,CUDA安装的版本是CUDA Toolkit 10.0,CUDNN是要与CUDA版本对应的。
最近安装了新发布的Deepin 20系统,界面确实漂亮些,且适配了很常用应用,Windows10和Deepin双系统体验相当不错。
由于学习需要配置Deep Learning环境要用到Tensorflow-gpu,首先得安装闭源驱动和cuda。
下面是本文是关于tensorflow-gpu和CUDA以及cuDNN匹配安装的记录。
一、安装NVIDIA Driver
1、下载驱动
去NVIDIA官方下载对应的驱动
GeForce xx series 是台式机显卡驱动
GeForce xx series (Notebooks) 是笔记本驱动
比如我这台老机器是GTX1050,我就按照下图这样设置
2、禁用NVIDIA开源驱动nouveau
新建一个黑名单让系统启动的时候不去加载开源驱动
sudo dedit /etc/modprobe.d/blacklist.conf
写入以下内容:
blacklist nouveau
options nouveau modeset=0
按ESC键,输入 :wq! 保存
然后执行,
sudo update-initramfs -u
重启系统,重启后查看是否生效,
输入以下代码检查是否禁用成功
lsmod |grep -i nouveau
如果没有输出内容,证明禁用成功了。
3、安装NVIDIA闭源驱动
先检查之前时候安装过闭源驱动和大黄蜂方案,如果有最好卸载掉。
nvidia-smi
没输出内容证明没有安装过闭源驱动
(1)如果你之前使用过大黄蜂方案,需要删除大黄蜂所安装的软件,可以用以下命令:
dpkg -l | grep -i nvidia |