矩池云上怎么快速安装tensorRT

本篇内容主要讲解“矩池云上怎么快速安装tensorRT”,感兴趣的朋友不妨来看看。本文介绍的方法操作简单快捷,实用性强。下面就让小编来带大家学习“矩池云上怎么快速安装tensorRT”吧!

创新互联公司科技有限公司专业互联网基础服务商,为您提供温江服务器租用高防服务器,成都IDC机房托管,成都主机托管等互联网服务。

国内镜像

https://mirrors.cloud.tencent.com/nvidia-machine-learning/ubuntu1804/x86_64/

检查系统版本

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/install/gpu

矩池云上怎么快速安装tensorRT

# 这里以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

到此,相信大家对“矩池云上怎么快速安装tensorRT”有了更深的了解,不妨来实际操作一番吧!这里是创新互联网站,更多相关内容可以进入相关频道进行查询,关注我们,继续学习!


当前名称:矩池云上怎么快速安装tensorRT
当前URL:http://cdiso.cn/article/piicgi.html

其他资讯