1. 制作dockerfile
在ubuntu主机下新建一个Dockerfile并且输入以下指令
# 使用 Ubuntu 18.04 作为基础镜像
FROM ubuntu:18.04
# 镜像作者信息
LABEL maintainer="1642942273@qq.com"
# 可以在此处添加其他自定义标签和元数据
# 定义容器启动时执行的命令(这里为空,即没有具体的执行命令)
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'