首页docker ubuntu安装cuda

docker ubuntu安装cuda

时间: 2025-04-24 22:07:10 浏览: 24

### 在 Ubuntu Docker 容器中安装配置 CUDA 为了在 Ubuntu 的 Docker 容器内成功安装并配置 CUDA，可以遵循如下方法：创建一个基于 nvidia/cuda 镜像的新 Dockerfile 是一种有效的方式。此镜像已经包含了 CUDA 和 NVIDIA 驱动程序的基础设置[^1]。 ```dockerfile FROM nvidia/cuda:11.0-base # 设置工作目录 WORKDIR /usr/src/myapp # 复制当前文件夹的内容到容器内的 /usr/src/myapp 文件夹下 COPY . . # 安装其他必要的依赖项 RUN apt-get update && \ apt-get install -y --no-install-recommends \ build-essential \ cmake \ git \ vim \ && rm -rf /var/lib/apt/lists/* CMD ["/bin/bash"] ``` 构建上述 Dockerfile 所定义的自定义镜像之后，可以通过命令 `docker build -t my-cuda-app .` 来完成操作。这会下载官方 Nvidia 提供的基础 CUDA 映像，并在其上添加任何额外所需的软件包或工具。启动新创建的带有 GPU 支持的容器时，应该利用特定于 Nvidia Container Toolkit 的选项来确保主机上的 GPU 能够被容器访问。使用如下的指令运行容器： ```bash docker run --gpus all -it --rm --name cuda-container my-cuda-app ``` 这条命令指定了所有的 GPU 设备都应该分配给这个新的交互式的容器实例，并命名为 `cuda-container`。当不再需要该容器的时候，它会被自动移除(`--rm`)以节省空间和资源。一旦进入了正在运行中的容器内部环境，则可通过执行简单的测试脚本来验证 CUDA 是否正常工作以及确认其版本号: ```bash nvidia-smi nvcc --version ``` 如果一切顺利的话，以上两条命令应当返回有关已安装的NVIDIA驱动状态的信息以及编译器的具体细节。

阅读全文