制作一个容器

1. 制作dockerfile

在ubuntu主机下新建一个Dockerfile并且输入以下指令

# 使用 Ubuntu 18.04 作为基础镜像
FROM ubuntu:18.04

# 镜像作者信息
LABEL maintainer="[email protected]"

# 可以在此处添加其他自定义标签和元数据

# 定义容器启动时执行的命令（这里为空，即没有具体的执行命令）
CMD []

然后cd到包含这个Dockerfile的路径下输入

docker build -t trtpy .

查看镜像是否生成

docker images

trtpy                        latest                 de8148a882bc   12 days ago     63.2MB

2. 拉镜像进入容器

docker run --gpus all -it --name trtpy -v $(pwd):/app trtpy /bin/bash

我里面已经有了一个anaconda3

bash ./Anaconda3-2021.11-Linux-x86_64.sh

成功安装好了anaconda3了之后设置一下conda init
先找到这个conda在哪里

whereis conda

/path/to/conda init

然后把设置好trtpy, 拉取TensorRT，cuda, cudnn

trtpy get-env

神奇的操作

Environment info: 
  trtpyc0: py39trt8cuda115cudnn8
    python: 3.9
    cuda: 11.5.2
    cudnn: 8.3.2.44
    tensorRT: 8.2.2.26
    min driver linux: 495.29.05
    min driver windows: 496.13
    binary_url_prefix: http://zifuture.com:1556/fs/{
    
    key}/1.2.6/py39-trt8-cuda11.5-cudnn8
Done, Usage: 'import trtpy.init_default as trtpy'