使用docker进行的tensorrt安装记录

1. 根据显卡版本拉取官网镜像

官网镜像地址

docker pull nvidia/cuda:11.4.0-cudnn8-devel-ubuntu20.04

2. 运行镜像/创建容器

docker run -it --name lhb_trt --gpus all -v /home/data/hablee_data_dir/trt_02:/home/trt_02 nvidia/cuda:11.4.0-cudnn8-devel-ubuntu20.04 /bin/bash

说明:

  1. –name trt_test02: 容器取名字叫trt_test02
  2. -v /home/data/hablee_data_dir/trt_02:/home/trt_02: 冒号前面的是本地路径(提前创建好),后面是映射的docker内的路径。以后本地路径和docker内部路径的内容都会随着对方的更改而更改(最好是以在本地修改为主,docker的命令行修改或创建的内容可能在外部没有权限访问)。
  3. nvidia/cuda:11.4.0-cudnn8-devel-ubuntu20.04: 拉取的镜像名称。
  4. /bin/bash: 以命令行方式进入docker

你可能感兴趣的:(深度学习,docker,容器,运维)