矩池云 · 2021年02月19日

矩池云上如何快速安装tensorRT

国内镜像

https://mirrors.cloud.tencent...

检查系统版本

source /etc/os-release && echo $VERSION_ID

导入apt仓库和其GPG key

# 第一步若输出 18.04,请用下面的命令
curl -fsSL https://mirrors.cloud.tencent.com/nvidia-machine-learning/ubuntu1804/x86_64/7fa2af80.pub | apt-key add -
echo "deb https://mirrors.cloud.tencent.com/nvidia-machine-learning/ubuntu1804/x86_64//" > /etc/apt/sources.list.d/cuda.list

安装所需版本的tensorRT

此次参照TensorFlow的官方文档

https://www.tensorflow.org/in...

# 这里以7.1.3版本为例
apt-get install -y --no-install-recommends libnvinfer7=7.1.3-1+cuda11.0 \
    libnvinfer-dev=7.1.3-1+cuda11.0 \
    libnvinfer-plugin7=7.1.3-1+cuda11.0
推荐阅读
关注数
1073
内容数
18
矩池云国内领先的GPU云共享平台,提供高性能的GPU租赁服务,助力人工智能、影视渲染行业。
目录
极术微信服务号
关注极术微信号
实时接收点赞提醒和评论通知
安谋科技学堂公众号
关注安谋科技学堂
实时获取安谋科技及 Arm 教学资源
安谋科技招聘公众号
关注安谋科技招聘
实时获取安谋科技中国职位信息