NVIDIA Docker是否需要安装CUDA？

Question

我正在设置用于机器学习开发的环境，并且考虑使用Docker。

是否需要在我的机器上安装Nvidia CUDA和/或CUdnn，或者仅在Docker容器中存在它才能工作？

提前感谢您的回答！

Answer 1

您需要：

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

通常不需要在基础计算机上安装CUDA（工具包）或CUDNN。这些可以放在容器中以便在容器中使用。

有关具体的安装说明，请参见here。