Hi 游客

更多精彩,请登录!

比特池塘 区块链茶馆 正文

矩池云上如何快速安装tensorRT

感恩965
70 0 0
国内镜像

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

# 这里以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
BitMere.com 比特池塘系信息发布平台,比特池塘仅提供信息存储空间服务。
声明:该文观点仅代表作者本人,本文不代表比特池塘立场,且不构成建议,请谨慎对待。
您需要登录后才可以回帖 登录 | 立即注册

本版积分规则

成为第一个吐槽的人

感恩965 小学生
  • 粉丝

    0

  • 关注

    0

  • 主题

    7