Linux(ubuntu22.04)系统下安装TensorRT、cuda、cudnn环境配置,多个版本安装共存

在Ubuntu 22.04系统上安装TensorRT、CUDA和cuDNN,并确保多个版本的共存,可以按照以下步骤进行:

  1. 安装CUDA:

    • 访问NVIDIA官网查找并下载相应版本的CUDA安装包。
    • 安装CUDA,例如:sudo sh cuda_<version>_linux.run
  2. 安装cuDNN:

    • 访问NVIDIA官网或者官方GitHub仓库下载cuDNN库。
    • 解压下载的cuDNN压缩包,并将相关文件复制到CUDA安装目录下,例如:

      
      
      
      sudo cp -P cuda/include/cudnn*.h /usr/local/cuda/include
      sudo cp -P cuda/lib64/libcudnn* /usr/local/cuda/lib64
      sudo chmod a+r /usr/local/cuda/include/cudnn*.h /usr/local/cuda/lib64/libcudnn*
  3. 安装TensorRT:

    • 从NVIDIA官网下载对应版本的TensorRT tar文件。
    • 解压TensorRT压缩包,并进入解压后的目录。
    • 运行TensorRT安装脚本,例如:

      
      
      
      cd TensorRT-<version>/
      sudo python3 install.py

在安装过程中,确保CUDA和cuDNN的版本与TensorRT版本兼容。如果需要安装不同版本的TensorRT或CUDA,可以创建多个虚拟环境,在每个虚拟环境中安装特定版本的软件包,例如使用virtualenvconda

注意:具体的安装命令和步骤可能会随着软件版本更新而变化,请参考官方文档以获取最新信息。

最后修改于:2024年08月23日 12:59

评论已关闭

推荐阅读

DDPG 模型解析,附Pytorch完整代码
2024年11月24日
DQN 模型解析,附Pytorch完整代码
2024年11月24日
AIGC实战——Transformer模型
2024年12月01日
Socket TCP 和 UDP 编程基础(Python)
2024年11月30日
python , tcp , udp
如何使用 ChatGPT 进行学术润色?你需要这些指令
2024年12月01日
AI
最新 Python 调用 OpenAi 详细教程实现问答、图像合成、图像理解、语音合成、语音识别(详细教程)
2024年11月24日
ChatGPT 和 DALL·E 2 配合生成故事绘本
2024年12月01日
omegaconf,一个超强的 Python 库!
2024年11月24日
【视觉AIGC识别】误差特征、人脸伪造检测、其他类型假图检测
2024年12月01日
[超级详细]如何在深度学习训练模型过程中使用 GPU 加速
2024年11月29日
Python 物理引擎pymunk最完整教程
2024年11月27日
MediaPipe 人体姿态与手指关键点检测教程
2024年11月27日
深入了解 Taipy:Python 打造 Web 应用的全面教程
2024年11月26日
基于Transformer的时间序列预测模型
2024年11月25日
Python在金融大数据分析中的AI应用(股价分析、量化交易)实战
2024年11月25日
AIGC Gradio系列学习教程之Components
2024年12月01日
Python3 `asyncio` — 异步 I/O,事件循环和并发工具
2024年11月30日
llama-factory SFT系列教程:大模型在自定义数据集 LoRA 训练与部署
2024年12月01日
Python 多线程和多进程用法
2024年11月24日
Python socket详解,全网最全教程
2024年11月27日