因为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安装（一）进行安装配置环境
//www.greatytc.com/p/32e2d395f247
4. 把docker保存成镜像
docker commit [OPTIONS] CONTAINER [REPOSITORY[:TAG]]
OPTIONS说明：
-a :提交的镜像作者；
-c :使用Dockerfile指令来创建镜像；
-m :提交时的说明文字；
-p :在commit时，将容器暂停。
先挖个坑，dockerfile有空我会上传
