NVIDIA Docker 需要安装 CUDA 吗?

Does NVIDIA Docker need CUDA installed?

我正在为机器学习开发设置我的环境,我想使用 Docker。

Nvidia CUDA and/or CUdnn 是否需要安装在我的机器上,还是只有存在于 docker 容器中才能工作?

提前感谢您的回答!

你需要:

  • nvidia 驱动程序
  • docker-ce 的最新版本(19.03 或更高版本)
  • nvidia container toolkit 也称为“nvidia docker”

您通常不需要在基础计算机上安装 CUDA(工具包)或 CUDNN。那些可以在容器中用于容器中。

有关具体安装说明,请参阅 here