Ubuntu 18.04 安装显卡驱动+Tensorflow

最新推荐文章于 2023-05-01 23:53:07 发布

Mountain Q

最新推荐文章于 2023-05-01 23:53:07 发布

阅读量403

点赞数

本文链接：https://blog.csdn.net/qck15570179927/article/details/103087333

版权

sudo vim /etc/modprobe.d/blacklist.conf（不熟悉vim的可以用sudo gedit /etc/modprobe.d/blacklist.conf）

在文本最后添加

blacklist nouveau
options nouveau modeset=0

然后更新之后重启

sudo update-initramfs -u
reboot

重启之后检查nouveau是否已经禁用

lsmod | grep nouveau

如果没有输出，表示禁用成功

使用快捷键Ctrl + Alt + F3进入命令行模式

首先需要关闭X-Window服务，否则可能无法安装显卡驱动

sudo service lightdm stop

注：如果你的电脑可能没有安装lightdm，而是使用gdm3图形界面，这时候你需要执行 sudo service gdm3 stop。当然这一步可以不用关gdm3，直接安装显卡驱动，我试过是没有问题的，关掉反而出现了问题，若出现问题输入（Alt + SysRq + reisub重启即可）

然后执行以下三条命令卸载原有的显卡驱动（这里已刚下载的NVIDIA-Linux-x86_64-418.43.run安装包为例）：

sudo apt-get remove --purge nvidia*
sudo chmod +x NVIDIA-Linux-x86_64-418.43.run
sudo ./NVIDIA-Linux-x86_64-418.43.run --uninstall

在终端中进入到NVIDIA-Linux-x86_64-418.43.run安装包所在的目录下，执行以下命令即可成功安装，安装中一直默认即可

sudo ./NVIDIA-Linux-x86_64-418.43.run

安装完之后测试一下安装是否成功

nvidia-smi

有输出说明安装成功
在这里插入图片描述
然后重启X-Window服务，最后重启电脑

sudo service lightdm start （或 sudo service gdm3 stop）
reboot

使用conda安装tensorflow之前先安装anaconda3

（1）打开终端（Ctrl + Alt + T）
（2）进入到下载位置

cd      文件下载路径

（3）在终端输入，运行 .sh文件（以Anaconda3-5.2.0-Linux-x86_64.sh为例）

bash  Anaconda3-5.2.0-Linux-x86_64.sh

在安装的过程中只要输入yes或按enter键就可以了
注：提示信息“Do you wish to proceed with the installation of Microsoft VSCode? [yes|no]”，输入no；
最后，重启终端，即可使用Anaconda3了；

conda install tensorflow-gpu==1.8.0（1.8.0为版本号，不写默认安装最新版）

补充：cudatoolkit和cudnn的卸载与重装

卸载
conda uninstall cudnn
conda uninstall cudatoolkit

重装
conda install cudatoolkit=9.0
conda install cudnn

（2）测试是否安装成功

import tensorflow as tf
tf.__version__

没有错误即会返回tensorflow的版本

（1）新建环境

conda create -n <env_name> python=3.6 # 新建环境 （3.6为创建环境python的版本，env_name为环境名，env_name文件可以在Anaconda安装目录envs文件下找到）

（2）激活环境

conda activate <env_name>

（3）安装tensorflow-gpu

conda install tensorflow-gpu

（4）退出激活环境

conda deactivate <env_name>

补充：可以指定url设置为国内镜像源提高下载速度,可参考这篇

conda config --add channels <channel_name or url>

关注