Docker容器中安装TensorRT8.0.0.3以及onnx2trt
Docker容器内部构建tensorRT过程记录一下自己的实现过程。一、基础环境基于:platform_pytorch:1.5_py37_v2.0 (或者dockerhub上的其他基础镜像)以及在Dockefile里面写了一些基础的依赖包的版本二、启动容器通过挂载的方式进行创建一个容器,指令如下nvidia-docker run --shm-size 12G -it --rm -v /balbala/xiaomoxian/liuyuankai/:/liuyuankai --name l
原创
2021-08-31 11:27:13 ·
1595 阅读 ·
1 评论