GPU A10 driver, CUDA 安装
GPU A10 driver, CUDA 安装
环境
HOST: ubuntu804LTS
GUST: 通过PCIE 透卡， KVM 虚拟机: ubuntu1804LTS
在gust 里面安装GPU driver
driver 安装 步骤
NVIDIA Telsa GPU 的 Linux 驱动在安装过程中需要编译 kernel module，系统需提前安装 gcc 和编译 Linux Kernel Module 所依赖的包，例如 kernel-devel-$(uname -r) 等。
- 执行以下命令，并输入 root 用户密码，切换至 root 用户
su
- 执行以下命令，查看当前系统中是否已安装 dkms。
dpkg -l | grep -i dkms
若有返回结果，则表明安装成功。若返回结果为空，则表明未安装 dkms，执行以下命令进行安装。
执行下面命令安装
apt-get install dkms
- 前往 Official Drivers 页面，按需选择 GRID 驱动类型。
如有填写个人信息的页面可选择直接跳过，当出现以下页面时，右键单击 AGREE&DOWNLOAD 并选择菜单中的复制链接地址。如下图所示：
- ssh 登录 GPU gust 机器, 使用 wget 命令，粘贴 步骤5 中复制的链接地址，下载安装包。
wget driver_url
- 执行以下命令，修改安装包权限。请将命令中的 xxx 替换为您实际的驱动版本号。
chmod +x NVIDIA-Linux-x86_64-xxxx.run
- 由于 NVIDIA 动的安装需要依赖 gcc 和 linux-kernel-headers，请依次执行以下命令，检查当前系统中是否已安装 gcc 和 kernel-devel 包。
dpkg -l | grep -i gcc
dpkg -l | grep -i linux-headers
如返回不为空， 表明已经安装成功过； 若返回结果为空，则表明未安装，执行以下命令进行安装。
sudo apt-get install gcc linux-kernel-headers
- 执行命令安装驱动程序，根据提示进行后续操作。请将命令中的 xxx 替换为您实际的驱动版本号。
sudo sh NVIDIA-Linux-x86_64-xxxx.run --ui=none --disable-nouveau --no-install-libglvnd --dkms -s
- 安装完成后，执行以下命令进行验证。
nvidia-smi
如返回信息类似下图中的 GPU 信息，则说明驱动安装成功。
安装最后出现 nvidia-drm 模块问题
Unable to load the “nvidia-drm” kernel module
ERROR: Installation has failed. Please see the file '/var/log/nvidia-installer.log' for details. You may find suggestions on fixing installation problems in the README available on the Linux driver download page at www.nvidia.com.
解决办法：
- 重新装
- https://onlycaptain.github.io/2018/08/18/Ubuntu下Nvidia驱动安装/
- https://www.machunjie.com/trouble/code_error/789.html
CUDA 安装
nvidia-smi 查看 driver 对应 cuda 版本
root@123:~# nvidia-smi
Fri Sep 16 16:27:55 2020
+-----------------------------------------------------------------------------+
| NVIDIA-SMI 470.141.03 Driver Version: 470.141.03 CUDA Version: 11.4 |
|-------------------------------+----------------------+----------------------+
| GPU Name Persistence-M| Bus-Id Disp.A | Volatile Uncorr. ECC |
| Fan Temp Perf Pwr:Usage/Cap| Memory-Usage | GPU-Util Compute M. |
| | | MIG M. |
|===============================+======================+======================|
| 0 NVIDIA A10 Off | 00000000:98:00.0 Off | 0 |
| 0% 29C P8 19W / 150W | 4MiB / 22731MiB | 0% Default |
| | | N/A |
+-------------------------------+----------------------+----------------------+
+-----------------------------------------------------------------------------+
| Processes: |
| GPU GI CI PID Type Process name GPU Memory |
| ID ID Usage |
|=============================================================================|
| |
+-----------------------------------------------------------------------------+
查看driver 对应的cuda 版本
https://docs.nvidia.com/cuda/cuda-toolkit-release-notes/index.html
https://developer.nvidia.com/cuda-toolkit-archive
https://developer.nvidia.com/cuda-11-4-0-download-archive?target_os=Linux&target_arch=x86_64&Distribution=Ubuntu&target_version=18.04&target_type=runfile_local
下载cuda-toolkit
wget https://developer.download.nvidia.com/compute/cuda/11.4.0/local_installers/cuda_11.4.0_470.42.01_linux.run
sudo chmod +x cuda_11.4.0_470.42.01_linux.run
./cuda_11.4.0_470.42.01_linux.run
安装选项，因为已经提前安装了显卡驱动，所以把显卡驱动的安装选项取消掉：
选择Install回车，出现如下界面，表示安装完成（不必担心警告，那是因为我们没有选择安装显卡驱动而出现的，忽略即可）：
测试一下：
root@123:~# nvcc --version
nvcc: NVIDIA (R) Cuda compiler driver
Copyright (c) 2005-2021 NVIDIA Corporation
Built on Wed_Jun__2_19:15:15_PDT_2021
Cuda compilation tools, release 11.4, V11.4.48
Build cuda_11.4.r11.4/compiler.30033411_0
配置CUDA环境
打开用户配置文件：
sudo vim ~/.bashrc
在文件中添加如下语句：
export PATH="/usr/local/cuda-11.4/bin:$PATH"
export LD_LIBRARY_PATH="/usr/lcoal/cuda-11.4/lib64:$LD_LIBRARY_PATH"
保存关闭后source文件使配置生效：
source ~/.bashrc
https://zhuanlan.zhihu.com/p/112138261
https://blog.csdn.net/weixin_38369492/article/details/107957296
