Linux(ubuntu22.04)系统下安装TensorRT、cuda、cudnn环境配置,多个版本安装共存
在Ubuntu 22.04系统上安装TensorRT、CUDA和cuDNN,并确保多个版本的共存,可以按照以下步骤进行:
安装CUDA:
- 访问NVIDIA官网查找并下载相应版本的CUDA安装包。
- 安装CUDA,例如:
sudo sh cuda_<version>_linux.run
。
安装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*
安装TensorRT:
- 从NVIDIA官网下载对应版本的TensorRT tar文件。
- 解压TensorRT压缩包,并进入解压后的目录。
运行TensorRT安装脚本,例如:
cd TensorRT-<version>/ sudo python3 install.py
在安装过程中,确保CUDA和cuDNN的版本与TensorRT版本兼容。如果需要安装不同版本的TensorRT或CUDA,可以创建多个虚拟环境,在每个虚拟环境中安装特定版本的软件包,例如使用virtualenv
或conda
。
注意:具体的安装命令和步骤可能会随着软件版本更新而变化,请参考官方文档以获取最新信息。
评论已关闭