linux的docker容器安装cuda

在Linux系统中，安装CUDA并在Docker容器中使用CUDA可以通过以下步骤完成：

1. 首先，确保您的Linux系统已经安装了Docker。您可以通过运行以下命令来检查Docker是否已经安装：

   docker --version

2. 如果您的系统尚未安装Docker，请根据您的Linux发行版选择适当的安装方法进行安装。您可以参考Docker官方文档以获取详细的安装指南。

3. 安装NVIDIA Docker运行时（nvidia-docker）。NVIDIA Docker运行时是一个用于在Docker容器中使用NVIDIA GPU的工具。您可以按照以下步骤进行安装：
- 添加NVIDIA Docker运行时的APT存储库：

     distribution=$(. /etc/os-release;echo $ID$VERSION_ID)
     curl -s -L https://nvidia.github.io/nvidia-docker/gpgkey | sudo apt-key add -
     curl -s -L https://nvidia.github.io/nvidia-docker/$distribution/nvidia-docker.list | sudo tee /etc/apt/sources.list.d/nvidia-docker.list
     sudo apt-get update

- 安装nvidia-docker2软件包：

     sudo apt-get install -y nvidia-docker2
     sudo systemctl restart docker

4. 确保您的系统上已经安装了适当版本的NVIDIA驱动程序。您可以通过运行以下命令来检查NVIDIA驱动程序的安装情况：

   nvidia-smi

5. 创建一个Dockerfile，用于构建包含CUDA的Docker镜像。以下是一个示例Dockerfile的内容：

   FROM nvidia/cuda:版本号
   # 安装其他依赖库和工具
   RUN apt-get update && apt-get install -y 其他依赖库和工具

6. 构建Docker镜像。在包含Dockerfile的目录中运行以下命令来构建镜像：

   docker build -t 镜像名称 .

7. 运行Docker容器并挂载CUDA设备。使用以下命令来运行容器：

   docker run --gpus all -it --name 容器名称 镜像名称

现在，您已经在Docker容器中成功安装了CUDA，并可以在容器中使用CUDA进行开发和计算。