andrewpsp/cuda-9.1

andrewpsp/cuda-9.1 镜像说明

镜像概述

本镜像为预装NVIDIA CUDA 9.1及相关驱动、运行时库的容器环境，需依赖主机已安装的NVIDIA驱动、Docker及nvidia-docker插件，实现容器对主机GPU资源的直接访问，支持CUDA 9.1版本应用的开发与运行。

核心功能

1. 包含CUDA 9.1完整工具包、运行时库、编译器及390.X版本驱动
2. 预设环境变量配置CUDA路径（LIBRARY_PATH、PATH、LD_LIBRARY_PATH等）
3. 支持通过nvidia-docker插件访问主机GPU资源
4. 提供cuda-runtime-9-1、cuda-libraries-9-1等组件，满足不同场景需求

使用场景

1. CUDA 9.1版本深度学习模型训练与推理
2. 基于GPU的科学计算任务
3. CUDA应用程序的开发与测试
4. 需要GPU加速的高性能计算场景

配置说明

前置依赖

• 主机需安装NVIDIA驱动（版本需兼容390.X）
• 主机需安装Docker及nvidia-docker插件（参考链接：https://github.com/NVIDIA/nvidia-docker/wiki%E3%80%81https://github.com/NVIDIA/nvidia-container-runtime%EF%BC%89

容器内环境变量

LIBRARY_PATH = /usr/local/cuda-9.1/lib64/stubs:${LIBRARY_PATH}
PATH= /usr/local/cuda-9.1/bin${PATH:+:${PATH}}
LD_LIBRARY_PATH=/usr/local/cuda-9.1/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}
CUDA_CUDART_LIBRARY=/usr/local/cuda-9.1/lib64/libcudart.so

部署示例

Docker Run命令

bash
nvidia-docker run --rm andrewpsp/cuda-9.1 nvidia-smi

该命令会启动容器并执行nvidia-smi，验证GPU资源是否正常访问。

Docker Compose示例（docker-compose.yml）

yaml
version: '3'
services:
  cuda-app:
    image: docker.xuanyuan.run/andrewpsp/cuda-9.1
    runtime: nvidia
    environment:
      - NVIDIA_VISIBLE_DEVICES=all
    command: nvidia-smi

使用docker-compose up启动服务，验证GPU访问。 ===FULL_DESC===