下面记录在Ubuntu20.04环境下,使用ASUS ATS-RTX4060-O8G-V2显卡,搭建Nvidia TensorRT开发环境。
1.安装步骤
0)准备工作
使用如下命令创建我们的工作目录:
mkdir ~/nvidia
再使用如下命令进入到上面的目录(接下来的步骤,如无特殊说明,均在该目录下进行):
cd ~/nvidia
1)安装CUDA
下载并安装NVIDIA CUDA Toolkit:
wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2004/x86_64/cuda-ubuntu2004.pin
sudo mv cuda-ubuntu2004.pin /etc/apt/preferences.d/cuda-repository-pin-600
wget https://developer.download.nvidia.com/compute/cuda/12.5.0/local_installers/cuda-repo-ubuntu2004