docker安装cuda

要在Docker容器中安装CUDA，您需要执行以下步骤：

1. 首先，确保您的主机上已经安装了NVIDIA显卡驱动程序。您可以通过运行以下命令来验证驱动程序是否正确安装：

   $ nvidia-smi

如果能够正确显示NVIDIA显卡信息，则表示驱动程序已经安装成功。

2. 接下来，您需要安装Docker。您可以根据您的操作系统选择适当的Docker版本并按照官方文档进行安装：https://docs.docker.com/engine/install/

3. 安装完Docker后，您需要下载NVIDIA Container Toolkit。这是一个用于在Docker容器中运行CUDA应用程序的工具集。您可以按照官方文档中的说明进行安装：https://docs.nvidia.com/datacenter/cloud-native/container-toolkit/install-guide.html#docker

4. 安装完成后，您可以使用以下命令来验证是否成功安装了NVIDIA Container Toolkit：

   $ docker run --gpus all nvidia/cuda:11.0-base nvidia-smi

如果能够正确显示NVIDIA显卡信息，则表示NVIDIA Container Toolkit已经成功安装。

5. 现在，您可以使用带有CUDA支持的基础映像来构建自己的Docker容器。例如，您可以使用以下命令来构建一个基于CUDA 11.0的容器：

   $ docker build -t my_cuda_app --build-arg CUDA_VERSION=11.0 .

这将在当前目录中的Dockerfile中构建一个名为"my_cuda_app"的容器，其中包含CUDA 11.0。

6. 构建完成后，您可以使用以下命令来运行您的CUDA应用程序：

   $ docker run --gpus all my_cuda_app

这将在新的容器中运行您的CUDA应用程序，并且能够正确地访问和使用GPU。

请注意，这只是一个简单的示例，您可能需要根据您的实际需求进行相应的调整和配置。此外，确保您的显卡和CUDA版本与指定的Docker映像和驱动程序版本兼容。