tensorrt环境探索

2024-02-23 09:55:58

tensorrt环境记录

版本	cuda version	drive version	GPU型号	显存大小	docker镜像	主要版本	运行状态	分析
0.1	11.2	450.51.06	tesla T4	15109MB	nnvcr.io/nvidia/pytorch:21.02-py3	torch=1.8.0 tensorrt=7.2.2.3 onnx=1.7.0	不稳定	CUDA版本与显卡驱动不兼容
0.2	11.2	450.51.06	tesla T4	15109MB	nnvcr.io/nvidia/tensorrt:21.02-py3	torch=1.8.0 tensorrt=7.2.2.3 onnx=1.7.0	不稳定	当前环境没有匹配的pytorch
0.3	11.2	450.51.06	tesla T4	15109MB	nnvcr.io/nvidia/pytorch:20.12-py3	torch=1.8.0 tensorrt=7.2.2.1 onnx=1.7.0	稳定
0.4	11.2	450.51.06	tesla T4	15109MB	nnvcr.io/nvidia/triton:21.02-py3	torch=1.8.0 tensorrt=7.2.2.3 onnx=1.7.0	稳定	所有需要的版本都要自己安装，比较麻烦