在 Dockerfile 中,可以通过 DEVICE
指令来指定设备。该指令用于将主机上的设备映射到容器内部。
例如,如果要在容器中使用 GPU 设备,可以使用以下方式:
FROM ubuntu:latest
# 安装所需的软件和驱动程序
RUN apt-get update && apt-get install -y --no-install-recommends \
cuda-toolkit-10-0 \
&& rm -rf /var/lib/apt/lists/*
# 设置环境变量
ENV CUDA_VISIBLE_DEVICES=0
# 将主机上的 NVIDIA 设备映射到容器内部
DEVICE /dev/nvidia*
上述示例中,首先安装了 CUDA 工具包和驱动程序。然后,通过设置 CUDA_VISIBLE_DEVICES
环境变量来指定可见的 GPU 设备(这里只有一个设备)。最后,使用 DEVICE
指令将 /dev/nvidia*
映射到容器内部。
请注意,在运行此 Docker 镜像时,需要确保宿主机上已正确安装并配置了 NVIDIA 驱动程序和 CUDA 工具包。
内容由零声教学AI助手提供,问题来源于学员提问