ubuntu20.04安装显卡驱动、CUDA11.6、cudnn8.4、anaconda
1、安装显卡驱动
打开终端输入下面命令查询需要安装的驱动版本
lspci | grep -i vga
以我的电脑为例,运行上面的命令后会输出
这里的2484对应的是GeForce RTX 3070型号的显卡。
查询显卡型号网址:https://admin.pci-ids.ucw.cz/mods/PC/10de?action=help?help=pci
会出现查询结果如下图所示
查询到显卡型号之后,去nvidia官网下载对应的显卡驱动(需要手动选择对应的类型噢)
网址:https://www.nvidia.com/Download/index.aspx
下载完成之后得到一个名为NVIDIA-Linux-x86_64-550.107.02.run的文件
运行下面的命令进行显卡驱动的安装:
# 给下载的驱动赋予可执行权限
sudo chmod 777 NVIDIA-Linux-x86_64-550.107.02.run
# 安装
# -no-opengl-files:只安装驱动文件,不安装OpenGL文件
# -no-x-check:安装驱动时关闭X服务,不设置可能导致安装失败。
sudo ./NVIDIA-Linux-x86_64-550.107.02.run –no-opengl-files -no-x-check
后续安装时弹出的选择我依次选的是
The distribution-provided pre-install script failed! Are you sure you want to continue?
--- 选择continue installation
Nvidia's 32-bit compatibility libraries?
--- 选择NO
Would you like to run the nvidia-xconfigutility to automatically update your x configuration so that the NVIDIA x driver will be used when you restart x? Any pre-existing x confile will be backed up.
--- 选择NO
安装完成点击 OK
安装完成后重启电脑
reboot
打开终端输入
nvidia-smi
出现类似下图这种显示,说明显卡驱动安装成功
2、安装CUDA
建议安装的cuda版本不高于nvidia-smi中输出的版本,本例安装的是cuda11.6.2
首先去官网下载CUDA Toolkit,网址:https://developer.nvidia.com/cuda-toolkit-archive
选择对应的版本之后,按照官网的下载和安装步骤进行操作
安装完成之后配置环境变量,配置完之后记得source一下
echo 'export PATH=/usr/local/cuda-11.6/bin:$PATH' >> ~/.bashrc
source ~/.bashrc
最后输入下面的命令查看cuda是否安装成功,安装成功的话会打印安装的CUDA版本等信息
nvcc --version
3、安装cudnn
cudnn下载网址:https://developer.nvidia.com/rdp/cudnn-archive
选择合适的版本进行下载,本例下载的是cudnn-local-repo-ubuntu2004-8.4.0.27_1.0-1_amd64.deb
下载完成之后,输入下面的命令进行安装
dpkg -i cudnn-local-repo-ubuntu2004-8.4.0.27_1.0-1_amd64.deb
4、安装anaconda
下载方式1:anaconda官网下载地址:https://www.anaconda.com/download
下载方式2:anaconda国内镜像下载地址:https://mirrors.tuna.tsinghua.edu.cn/anaconda/archive/?C=M&O=A
找到需要的进行下载,本例用的是Anaconda3-2023.07-2-Linux-x86_64.sh
下载完成之后,打开终端,输入下面的命令进行安装
bash Anaconda3-2023.07-2-Linux-x86_64.sh
安装过程中有各种确认信息,直接回车,根据提示,同意licenses。最后会提示输入安装路径,默认安装路径就直接回车,想要安装在别的地方就修改路径后回车。直到安装结束。安装完成之后配置环境变量,配置完之后记得source一下
echo 'export PATH=/home/user/anaconda3/bin:$PATH' >> ~/.bashrc
source ~/.bashrc
可以在终端输入conda info,回车,查看相关安装信息