一、安装NVIDIA显卡驱动
1. 在官网下载对应的显卡驱动(根据自己显卡的类型选择对应的驱动)
https://www.geforce.cn/drivers
2.禁用自带的 nouveau开源驱动
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
如果没有输出,表示禁用成功
3. 删除旧的NVIDIA驱动
使用快捷键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
4. 安装新驱动
在终端中进入到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
二、安装Tensorflow
使用conda
安装tensorflow
之前先安装anaconda3
1. Anaconda安装
- 选择相应的anaconda版本进行下载(anaconda python 版本对应关系)
官网下载地址: https://www.anaconda.com/distribution/#linux
清华大学开源软件镜像站下载地址: https://mirrors.tuna.tsinghua.edu.cn/help/anaconda/
(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了;
2. Tensorflow的安装
- (1)这里使用conda安装tensorflow,在终端中输入
conda install tensorflow-gpu==1.8.0(1.8.0为版本号,不写默认安装最新版)
安装完之后,使用conda list
可以看到conda在安装tensorflow时会自动为系统安装cudatoolkit和cudnn
注:安装了 cudatoolkit 后,只要系统上存在与当前的 cudatoolkit 所兼容的 Nvidia 驱动,则已经编译好的 CUDA 相关的程序就可以直接运行,而不需要安装完整的 Nvidia 官方提供的 CUDA Toolkit。
补充:cudatoolkit和cudnn的卸载与重装
卸载
conda uninstall cudnn
conda uninstall cudatoolkit
重装
conda install cudatoolkit=9.0
conda install cudnn
(2)测试是否安装成功
import tensorflow as tf
tf.__version__
没有错误即会返回tensorflow的版本
- 另外可以新建一个python虚拟环境
(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>