因为Docker不能使用Nvidia驱动，所以请务必使用Nvidia-docker

1. 安装Nvidia-docker

官网有安装方法
https://docs.nvidia.com/datacenter/cloud-native/container-toolkit/install-guide.html

1. 下载软件源

distribution=$(. /etc/os-release;echo $ID$VERSION_ID) \
   && curl -s -L https://nvidia.github.io/nvidia-docker/gpgkey | sudo apt-key add - \
   && curl -s -L https://nvidia.github.io/nvidia-docker/$distribution/nvidia-docker.list | sudo tee /etc/apt/sources.list.d/nvidia-docker.list

2. 安装

sudo apt-get update
sudo apt-get install -y nvidia-docker2

3. 重启docker服务

sudo systemctl restart docker

4. 查看镜像和进入docker

docker images
docker run -it --gpus all nvidia/cuda:11.0-base

2. 重新安装cuda

虽然里面有cuda，文件不完整，为保证以后的时候，可以重新安装一次cuda，可以到cuda官网下载相应版本的run离线文件安装，这样最快
cuda release官网：https://developer.nvidia.com/cuda-toolkit-archive

3. 参照TVM安装（一）进行安装配置环境

https://www.jianshu.com/p/32e2d395f247

4. 把docker保存成镜像

docker commit [OPTIONS] CONTAINER [REPOSITORY[:TAG]]

OPTIONS说明：

-a :提交的镜像作者；

-c :使用Dockerfile指令来创建镜像；

-m :提交时的说明文字；

-p :在commit时，将容器暂停。

先挖个坑，dockerfile有空我会上传