一键卸载和安装 nvidia、cuda、cudnn、tensorrt

最新推荐文章于 2024-04-03 15:14:32 发布

wohu007

最新推荐文章于 2024-04-03 15:14:32 发布

阅读量391

点赞数 1

分类专栏： Ubuntu 文章标签： ubuntu 内核

本文链接：https://blog.csdn.net/wohu1104/article/details/136697250

版权

Ubuntu 专栏收录该内容

31 篇文章 8 订阅

订阅专栏

1. 卸载 nvidia、cuda、cudnn、tensorrt

sudo apt purge  \
  "cuda*"             \
  "libcudnn*"         \
  "libnvinfer*"       \
  "libnvonnxparsers*" \
  "libnvparsers*"     \
  "tensorrt*"         \
  "nvidia*"

sudo apt autoremove

2. 安装 nvidia、cuda、cudnn、tensorrt

os=ubuntu2204
cuda_version=12.3
cudnn_version=8.9.5.29-1+cuda12.2
tensorrt_version=8.6.1.6-1+cuda12.0

wget https://developer.download.nvidia.com/compute/cuda/repos/$os/$(uname -m)/cuda-keyring_1.1-1_all.deb
sudo dpkg -i cuda-keyring_1.1-1_all.deb
sudo apt-get update
cuda_version_dashed=$(eval sed -e "s/[.]/-/g" <<< "${cuda_version}")
sudo apt-get -y install cuda-${cuda_version_dashed}