Installing the NVIDIA Container Toolkit 容器配置GPU
网上的帖子都比较旧了,这是最新官方地址:
https://docs.nvidia.com/datacenter/cloud-native/container-toolkit/latest/install-guide.html
注意:除了安装,还有配置,不是说安装上就完事
都弄好之后,就可以docker run 了,下面是个例子
docker run --gpus all --ipc=host --ulimit memlock=-1 --ulimit stack=67108864 -it chenhsuanlin/neuralangelo:23.04-py3
bash