想请教一下大家。我在 Ubuntu 22.04 上用 Docker 跑一个 Python 项目，需要用到 CUDA。主机已经装好 NVIDIA driver，nvidia-smi 在主机端正常，但容器里面抓不到 GPU，PyTorch 一直显示 CUDA not available。我有安装 nvidia-container-toolkit，也重启过 doc...
Original
想請教一下大家。我在 Ubuntu 22.04 上用 Docker 跑一個 Python 專案，需要用到 CUDA。主機已經裝好 NVIDIA driver，nvidia-smi 在主機端正常，但容器裡面抓不到 GPU，PyTorch 一直顯示 CUDA not available。我有安裝 nvidia-container-toolkit，也重啟過 docker。請問還可能漏了哪個步驟？還是 docker run 需要特別參數？