准备工作
注:本机系统为 Ubuntu20.04
1. 安装显卡驱动
打开‘软件和更新,点击附加驱动安装显卡驱动。
2. gcc安装
若系统为Ubuntu22.04,则需要安装。
在终端输入一下指令,查看有没有gcc。
gcc --version
3. gcc安装
本机ubuntu20.04 gcc自带版本为9.4.0
安装gcc-7,command:
sudo apt-get install gcc-7 g++-7
安装完gcc-7,系统中就存在两个版本的gcc,因此要设置默认的gcc,命令如下:
sudo update-alternatives --install /usr/bin/gcc gcc /usr/bin/gcc-7 9
sudo update-alternatives --install /usr/bin/gcc gcc /usr/bin/gcc-9 1
此命令可以通过update-alternatives设置gcc各版本的优先级,优先级最高的为系统默认版本,可以用下述命令显示其优先级:
sudo update-alternatives --display gcc
CUDA安装
安装CUDA 11.7,选择runfile(local),并使用生成的指令进行下载和安装。
选择对应的版本,然后在下面就可以安装命令:
woet httos://developer.donnload,nvidia.com/compute/cuda/11,7.0/local installers/cuda 11,7.0 515,43.04 linux.rurs
sudo sh cuda 11.7.0 51543.04 linux.run
在终端执行以上生成命令,见下图。
若第1步提示Existing package manager installation of the driver found. It is strongly recommended that you remove this before continuing.,选择continue,在下一步中去除driver项,之后选择install:
安装完成后,显示如下:
在~/.bashrc文件中添加如下环境变量:
export PATH=/usr/local/cuda-11.7/bin${PATH:+:${PATH}}
export LD_LIBRARY_PATH=/usr/local/cuda-11.7/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}
如果之前安装过其他版本,将之前的环境变量删除,导入本次安装的环境变量。
注:vim操作 输入 i 插入 然后 esc 输入:wq保存
查看vim指令点击:链接。
使用一下命令更新系统环境变量
source ~/.bashrc
sudo ldconfig
至此cuda安装完成,输入nvcc -V命令查看cuda信息
nvcc -V
cudnn安装
下载链接:https://developer.nvidia.com/rdp/cudnn-download
注: 图中红框版本
更改usr/local/cuda/文件夹下面的include和lib64文件夹的权限。
常用的修改权限的指令
sudo chmod 600 ××× (只有所有者有读和写的权限)
sudo chmod 644 ××× (所有者有读和写的权限,组用户只有读的权限)
sudo chmod 700 ××× (只有所有者有读和写以及执行的权限)
sudo chmod 666 ××× (每个人都有读和写的权限)
sudo chmod 777 ××× (每个人都有读和写以及执行的权限)
执行修改文件夹权限的命令:
cd /usr/local/cuda
sudo chmod 666 include
sudo chmod 666 lib64
然后解压cudnn-linux-x86_64-8.6.0.163_cuda11-archive.tar.xz文件。等待解压完成后,在下载文件夹中右键打开终端。执行如下命令:
sudo cp cudnn-*-archive/include/cudnn*.h /usr/local/cuda/include
sudo cp cudnn-*-archive/lib/libcudnn* /usr/local/cuda/lib64
sudo chmod a+r /usr/local/cuda/include/cudnn*.h /usr/local/cuda/lib64/libcudnn*
见NVIDIA官方安装指导1.3.1(Installation Guide: Step-by-step instructions for installation and upgrade)
验证是否安装成功,执行命令:
sudo cat /usr/local/cuda/include/cudnn_version.h | grep CUDNN_MAJOR -A 2
至此cudnn也安装完成。