Ubuntu20.04+CUDA12.5+TensorRT10.2.0

原创已于 2024-11-12 19:12:43 修改 · 225 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#linux #ubuntu #运维

于 2024-10-31 21:36:24 首次发布

CUDA:

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.1/local_installers/cuda-repo-ubuntu2004-12-5-local_12.5.1-555.42.06-1_amd64.deb
sudo dpkg -i cuda-repo-ubuntu2004-12-5-local_12.5.1-555.42.06-1_amd64.deb
sudo cp /var/cuda-repo-ubuntu2004-12-5-local/cuda-*-keyring.gpg /usr/share/keyrings/
sudo apt-get update
sudo apt-get -y install cuda-toolkit-12-5
sudo apt-get install -y cuda-drivers

TensorRT:

wget https://developer.nvidia.com/downloads/compute/machine-learning/tensorrt/10.2.0/local_repo/nv-tensorrt-local-repo-ubuntu2004-10.2.0-cuda-12.5_1.0-1_amd64.deb
os="ubuntu2004"
tag="10.2.0-cuda-12.5"
sudo dpkg -i nv-tensorrt-local-repo-${os}-${tag}_1.0-1_amd64.deb
sudo cp /var/nv-tensorrt-local-repo-${os}-${tag}/*-keyring.gpg /usr/share/keyrings/
sudo apt-get update
sudo apt-get install tensorrt

其它

dpkg -l | grep tensorrt
sudo apt-get remove --purge tensorrt

#调整nvcc版本
export PATH=/usr/local/cuda-12.5/bin:$PATH
export LD_LIBRARY_PATH=/usr/local/cuda-12.5/lib64:$LD_LIBRARY_PATH

sudo apt-get --purge remove "cuda*"
dpkg -l | grep cuda
sudo dpkg --purge libcudart10.1:amd64
sudo dpkg --purge nvidia-cuda-dev
sudo dpkg --purge nvidia-cuda-doc
sudo dpkg --purge nvidia-cuda-gdb
sudo dpkg --purge nvidia-cuda-toolkit
sudo apt-get autoremove
sudo apt-get clean