nvidia-docker安装

1. Docker安装

https://docs.docker.com/engine/install/centos/

2. nvidia-docker配置

nvidia-docker wrapper
https://docs.nvidia.com/datacenter/cloud-native/container-toolkit/latest/install-guide.html

三、测试

nvidia-docker run --rm nvidia/cuda:12.1.0-base-ubuntu20.04 nvidia-smi
1

参考资料：
https://docs.nvidia.com/datacenter/cloud-native/container-toolkit/latest/install-guide.html