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 | 稳定 | 所有需要的版本都要自己安装,比较麻烦 |