docker nvidia

有时候需要比较麻烦的环境，可以使用nvidia适配好了的环境

nvidia docker的教程：

https://catalog.ngc.nvidia.com/orgs/nvidia/containers/pytorch

镜像列表（可以看cuda版本、torch版本、python版本等）：

https://docs.nvidia.com/deeplearning/frameworks/pytorch-release-notes/rel_21-07.html#rel_21-07

使用：

docker run --gpus all -itd nvcr.io/nvidia/pytorch:20.12-py3 #一定要有d，退出后还能后台运行

其它命令：

docker ps # 查看有哪些容器
docker exec -it CONTAINER ID /bin/bash # 进入容器
ctrl+D #退出容器
# 删除容器
docker stop <CONTAINER ID|NAME> #先停止要删除的容器
docker rm <CONTAINER ID|NAME> #删除