我可以只使用nvidia驱动主机的基于码头的系统？

Question

我很难理解库达和码头生态系统。

在主机(ubuntu22.04)服务器上，我想生成多个Machiene学习木星笔记本。

如果我只在主机ubuntu中安装这样的Nvidia驱动程序，是否就足够了：

sudo apt-get install linux-headers-$(uname -r)
DISTRIBUTION=$(. /etc/os-release;echo $ID$VERSION_ID | sed -e 's/\.//g')
echo $DISTRIBUTION
wget https://developer.download.nvidia.com/compute/cuda/repos/$DISTRIBUTION/x86_64/cuda-keyring_1.0-1_all.deb
sudo dpkg -i cuda-keyring_1.0-1_all.deb
sudo apt-get update
sudo apt-get -y install cuda-drivers

sudo reboot

#After reboot verify if the CUDA driver is installed:
nvidia-smi

然后在像这样的容器中安装cuda：

wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2204/x86_64/cuda-ubuntu2204.pin
sudo mv cuda-ubuntu2204.pin /etc/apt/preferences.d/cuda-repository-pin-600
sudo apt-key adv --fetch-keys https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2204/x86_64/3bf863cc.pub
sudo add-apt-repository "deb https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2204/x86_64/ /"
sudo apt-get update
sudo apt-get -y install cuda

我认为最简单的方法是只在主机系统上安装所需的软件包，并用所有必需的包充实容器。因此，我怀疑这种做法是否合理。

1. 我是否正确理解容器将使用来自主机系统的驱动程序？
2. 安装cuda是否足够安装在容器中，或者我将安装cuda-toolkit，因为它包含更多的附加包？

Answer 1

我是否正确理解容器将使用来自主机系统的驱动程序？

是的，容器将使用来自主机系统的驱动程序。如果您正在构建自己的容器，请不要在容器中安装驱动程序。

安装cuda是否足够安装在容器中，或者我将安装cuda工具包，因为它包含更多的附加包？

在这个场景中，您可能永远不想安装cuda。您可以在主机上安装cuda (这也将安装问题1中提到的驱动程序)。这是可以接受的。在容器中，如果您自己构建它，则不希望安装cuda，最多只能安装cuda-toolkit。

我可以只使用nvidia驱动主机的基于码头的系统？

除了linux安装之外，主机中还有一般需要3件物品，以便为启用CUDA的容器使用做好准备：

1. GPU驱动程序
2. 最近的码头版本
3. NVIDIA容器工具包(安装说明见上面的链接)

没有必要在主机上安装CUDA工具包(即GPU驱动程序安装之外的项目)。如果容器中需要它们，通常会将它们安装在容器中。