ubuntu 安装tensorRT报错nvrtc: error: invalid value for --gpu-architecture (-arch)

安装参考: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,下载

  • 4
    点赞
  • 4
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值