安装参考:https://blog.csdn.net/zong596568821xp/article/details/86077553
遇见问题:
nvrtc: error: invalid value for --gpu-architecture (-arch)
原因:
tensort与cudnn版本不一致导致
我的硬件:GeForce RTX 3090
cuda 11.0(好像3090至少要cuda11.0,否则又会出错)
报错时对应的tensorrt和cudnn版本:TensorRT-7.1.3.4,cudnn:8.0.1.13
解决:
升级为cuDNN:8.1.1,TensorRT 7.2.3
卸载cuDNN:
sudo rm -rf /usr/local/cuda/lib64/libcudnn*
sudo rm -rf /usr/local/cuda/include/cudnn*
#把新版本的cudnn库拷贝到系统中
sudo cp /home/xx/softwares/cuda/include/* /usr/local/cuda/include/
sudo cp /home/xx/softwares/cuda/lib64/* /usr/local/cuda/lib64/
卸载TensorRT :
pip uninstall tensorrt
cuDNN:8.1.1,下载
"""下载下来是solitairetheme8 格式"""
cp cudnn-11.2-linux-x64-v8.1.1.33.solitairetheme8 cudnn-11.2-linux-x64-v8.1.1.33.tgz
tar -zxvf cudnn-11.2-linux-x64-v8.1.1.33.tgz
TensorRT 7.2.3 for Linux and CUDA 11.0,下载