本文章的前提是已经安装好了Ubuntu系统,对应版本为ubuntu20.04,cuda-11.3,cudnn-8.6,torch-1.10.0, python-3.8
一、gcc安装
打开终端查看有没有安装gcc
# 终端输入rpm -q gcc
caoweiguang@caoweiguang-virtual-machine:~$ rpm -q gcc
如果没有,则执行如下命令,安装gcc
# 终端输入sudo apt update
caoweiguang@caoweiguang-virtual-machine:~$ sudo apt update
# 终端输入sudo apt install build-essential
caoweiguang@caoweiguang-virtual-machine:~$ sudo apt install build-essential
二、Nidia显卡驱动安装
1.卸载原有驱动
# 终端输入
sudo apt-get remove --purge nvidia*
2.禁用nouveau
终端输入如下命令
# 终端输入
sudo gedit /etc/modprobe.d/blacklist.conf
在打开的blacklist.conf末尾添加如下内容
blacklist nouveau
options nouveau modeset=0
保存好之后,在终端输入
sudo update-initramfs –u
重启系统,之后终端输入下面命令,若没有反应,则禁用成功
lsmod | grep nouveau
3.下载显卡驱动
在官方驱动 | NVIDIA可以查找适合自己的版本,我安装在笔记本上,所以是Notebooks版的
点击搜索
点击下载
下在完成后,进入NVIDIA下载保存的位置,选择在终端打开
执行如下指令
# 注意