如何在colab中安装TensorRT,这篇博文主要讲述,怎样白嫖谷歌colab的gpu,并安装TensorRT进行简单的推理。附notebook和安装包百度云连接,下载后可以直接运用。
一、主要步骤。1、注册谷歌账号并申请谷歌云盘;2、下载TensorRT的deb包和tar包;3、将下载好的两个安装包上传谷歌云盘;4、安装英伟达包;5、安装TensorRT的deb包;6、安装TensorRT的tar包;7、验证是否安装成功;8、运作简单demo。
二、具体细节。1、申请谷歌云盘。2、下载TensorRT的deb包和tar包,并上传到谷歌云盘,需要建一个文件夹存放deb包,tar包和ipthy脚本。3、打开脚本进去colab,挂载云盘,进去工作目录。4、卸载TensorFlow2,安装安装TensorFlow1.14。5、解压并安装deb包。6、安装依赖包。7、检查deb包是否安装成功。8、解压并安装tar包。9、检查tar包是否安装成功。
三、运行demo进行推理。



