docker - NVIDIA Docker 需要安装 CUDA 吗？
问题描述
我正在为机器学习开发设置环境，并考虑使用 Docker。
Nvidia CUDA 和/或 CUdnn 是否需要安装在我的机器上，还是只能通过存在于 docker 容器中才能工作？
提前感谢您的回答！
解决方案
你需要：
- 英伟达驱动
- 最新版本的 docker-ce（19.03 或更高版本）
- 和nvidia 容器工具包也称为“nvidia docker”
您通常不需要在基础机器上安装 CUDA（工具包）或 CUDNN。那些可以在容器中用于容器中。
有关具体安装说明，请参见此处。