duan8/docker/trt8/Dockerfile
haku e418cb9e66
add dockerfile trt8_cuda11 (#1010)
* add dockerfile trt8_cuda11

* update docker readme
2022-06-11 11:51:41 +08:00

5 lines
82 B
Docker

ARG TENSORRT="8"
ARG CUDA="11"
FROM hakuyyf/tensorrtx:trt${TENSORRT}_cuda${CUDA}