要讓 Nvidia Docker 容器訪問 GPU,首先需要確保您的系統上安裝了 Nvidia 驅動程序和 CUDA 工具包。然后,您可以按照以下步驟操作:
--gpus
標志來指定要分配給容器的 GPU 數量。例如,要分配一個 GPU:docker run --gpus all [其他選項] nvidia/cuda
nvidia-smi
命令來查看 GPU 信息。例如:nvidia-smi
通過以上步驟,您就可以讓 Nvidia Docker 容器訪問 GPU,并利用 GPU 加速性能進行計算任務。