笔记笔记
  • Home
  • AI&ML
  • Example
  • Zoo
  • 关于
⌘ K
Ubuntu
NVIDIA
Python 环境
systemctl
最后更新时间:
Copyright © 2023-2024 | Powered by dumi | GuoDapeng | 冀ICP备20004032号-1 | 冀公网安备 冀公网安备 13024002000293号

TABLE OF CONTENTS

‌
‌
‌
‌

NVIDIA

Ubuntu 22.04.4 LTS

参考资料

Ubuntu22.04安装CUDA和cuDNN详细过程记录

硬件

Intel® Core™ i5-12600K
msi 主板 PRO Z790-P WIFI
GeForce RTX 2080 Ti
32G 2T

安装驱动

下图是一般新安装系统的情况,没有正确地安装驱动,导致CUDA无法使用。

# 安装驱动
sudo ubuntu-drivers autoinstall

重启系统后,用nvidia-smi命令检查是否安装成功。

我重置了 BIOS,全盘格式化重装了系统,这个弹窗不见了

Enroll MOK -> continue -> yes,输入密码,选 reboot 重启即可。

安装 CUDA Toolkit

CUDA Toolkit Archive

下载合适版本的CUDA,我下载的是CUDA 12.0.0。

wget https://developer.download.nvidia.com/compute/cuda/12.0.0/local_installers/cuda_12.0.0_525.60.13_linux.run
sudo sh cuda_12.0.0_525.60.13_linux.run

这个输出比较重要。

Please make sure that
- PATH includes /usr/local/cuda-12.0/bin
- LD_LIBRARY_PATH includes /usr/local/cuda-12.0/lib64, or, add /usr/local/cuda-12.0/lib64 to /etc/ld.so.conf and run ldconfig as root

其实 /usr/local/cuda-12.0 会链接一个 /usr/local/cuda 位置。

# 在 ~/.bashrc 添加以下内容
export PATH=$PATH:/usr/local/cuda/bin
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda/lib64
# 执行 source ~/.bashrc 或者重启一个终端,就可以检查是否安装成功了。

安装 cuDNN

cuDNN Archive

这里我选择了Download cuDNN v8.9.1 (May 5th, 2023), for CUDA 12.x,下载了Local Installer for Linux x86_64 (Tar)。

NVIDIA cuDNN Documentation

这里有所有版本的文档。

cuDNN 8.9.1 Installation Guide

# Unzip the cuDNN package.
tar -xvf cudnn-linux-x86_64-8.x.x.x_cudaX.Y-archive.tar.xz
# Copy the following files into the CUDA toolkit directory.
sudo cp cudnn-*-archive/include/cudnn*.h /usr/local/cuda/include
sudo cp -P cudnn-*-archive/lib/libcudnn* /usr/local/cuda/lib64
sudo chmod a+r /usr/local/cuda/include/cudnn*.h /usr/local/cuda/lib64/libcudnn*
# 验证
cat /usr/local/cuda/include/cudnn_version.h | grep CUDNN_MAJOR -A 2

MAJOR、MINOR、PATCHLEVEL 依次是 8、9、1,与版本号对应。

安装 TensorRT

NVIDIA TensorRT documentation

选择合适的版本。

# Unzip the TensorRT package.
tar -xzvf TensorRT-${version}.Linux.${arch}-gnu.${cuda}.tar.gz
# 在 ~/.bashrc 添加以下内容
export PATH=/home/admin1/work/TensorRT-8.6.1.6/bin:$PATH
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/home/admin1/work/TensorRT-8.6.1.6/lib
export LIBRARY_PATH=/home/admin1/work/TensorRT-8.6.1.6/lib:$LIBRARY_PATH
# 执行 source ~/.bashrc 或者重启一个终端,就可以检查是否安装成功了。
# 验证
trtexec
# 验证
cd work/TensorRT-8.6.1.6/samples/sampleOnnxMNIST/
make
# 输出 Linking: ../../bin/sample_onnx_mnist 就成功了。
Preview
Preview
Preview
Preview
Preview
Preview
Preview
Preview
Preview
Preview
Preview
Preview
Preview
Preview
Preview
Preview
Preview
Preview
Preview
Preview