如何创建支持CUDA的Windows版Docker镜像以运行EXE程序？

核心问题修正 #

你之前的Dockerfile使用了Linux版CUDA镜像，这完全无法运行Windows平台的myapp.exe——Linux容器与Windows容器不兼容，必须基于Windows基础镜像构建。

操作步骤 #

1. 切换Docker到Windows容器模式
   打开Docker Desktop，在设置中切换至Windows容器（默认是Linux容器），这是运行Windows程序的必要前提。

2. 选择匹配的Windows基础镜像
   选用微软官方的Windows Server Core镜像，比如mcr.microsoft.com/windows/servercore:ltsc2019，版本尽量和你打包myapp.exe时的Windows系统一致，避免兼容性问题。

3. 在镜像中安装Windows版CUDA运行时
   前往NVIDIA官方获取对应版本的Windows CUDA运行时离线安装包，将下载地址填入Dockerfile的PowerShell命令中，通过静默安装完成配置，确保CUDA版本与你的程序依赖版本一致（比如12.3.1）。

4. 复制并启动你的EXE程序
   将myapp.exe复制到容器内，设置工作目录后配置启动命令。

正确的Dockerfile示例 #

# 选用Windows Server Core基础镜像
FROM mcr.microsoft.com/windows/servercore:ltsc2019

# 设置环境变量，指定CUDA版本（需与程序依赖版本一致）
ENV CUDA_VERSION=12.3.1

# 使用PowerShell下载并静默安装CUDA运行时
# 注意：将下方地址替换为对应版本Windows CUDA运行时的官方离线安装包地址
RUN powershell -Command \
    Invoke-WebRequest -Uri "此处替换为CUDA $env:CUDA_VERSION的Windows离线安装包下载地址" -OutFile cuda_install.exe; \
    Start-Process -FilePath cuda_install.exe -ArgumentList "-s", "--override" -Wait; \
    Remove-Item -Path cuda_install.exe -Force

# 复制你的EXE程序到容器内
COPY myapp.exe /app/
WORKDIR /app

# 设置启动命令
CMD ["myapp.exe"]

关键前置要求 #

• 宿主机必须安装对应版本的NVIDIA显卡驱动，驱动版本需支持你指定的CUDA版本；
• 安装NVIDIA Container Toolkit for Windows，确保Docker容器能访问宿主机的NVIDIA GPU；
• 打包myapp.exe时的CUDA依赖版本，必须与镜像中安装的CUDA运行时版本完全一致。

内容的提问来源于stack exchange，提问作者theateist