如何在我的 docker ubuntu 映像中获取 nvidia 驱动程序？
Posted
技术标签:
【中文标题】如何在我的 docker ubuntu 映像中获取 nvidia 驱动程序？【英文标题】：How do I get an nvidia driver in my docker ubuntu image? 【发布时间】：2020-09-21 10:45:15 【问题描述】：我需要在 kubernetes 上运行一个 cuda 二进制文件。我已将节点设置为将kubernetes nvidia device plugin 与 nvidia-docker2 一起使用。这是我的 Dockerfile：
FROM ubuntu:18.04
COPY addarrays /addarrays
ENTRYPOINT [ "/addarrays" ]
当我通过 nvidia-docker2 或 kubernetes 运行 docker 映像时，会出现以下错误：
Cuda failure addarrays.cu:9: 'CUDA driver version is insufficient for CUDA runtime version'
Cuda failure addarrays.cu:9: 'CUDA driver version is insufficient for CUDA runtime version'
Cuda failure addarrays.cu:9: 'CUDA driver version is insufficient for CUDA runtime version'
Cuda failure addarrays.cu:9: 'CUDA driver version is insufficient for CUDA runtime version'
Cuda failure addarrays.cu:9: 'CUDA driver version is insufficient for CUDA runtime version'
Cuda failure addarrays.cu:9: 'CUDA driver version is insufficient for CUDA runtime version'
addarrays: addarrays.cu:62: int main(): Assertion `hostArrayTmp[i] == hostArrayDest[i]' failed.
看起来我的 docker 映像需要一个 nvidia 驱动程序。我已经像这样修改了 Dockerfile：
FROM ubuntu:18.04
RUN apt install software-properties-common
RUN add-apt-repository ppa:graphics-drivers
RUN apt update
RUN apt install nvidia-driver-440
COPY addarrays /addarrays
ENTRYPOINT [ "/addarrays" ]
安装 add-apt-repository 需要 software-properties-common，但安装失败并显示以下消息：
E: Unable to locate package software-properties-common
我需要做什么才能在我的 docker 映像中安装 nvidia 驱动程序？
【问题讨论】：
【参考方案1】：您必须先执行apt update。在安装命令中，您应该使用 -y 标志。
FROM ubuntu:18.04
RUN apt update
RUN apt install software-properties-common -y
RUN add-apt-repository ppa:graphics-drivers
RUN apt install nvidia-driver-440 -y
COPY addarrays /addarrays
ENTRYPOINT [ "/addarrays" ]
【讨论】：
以上是关于如何在我的 docker ubuntu 映像中获取 nvidia 驱动程序？的主要内容，如果未能解决你的问题，请参考以下文章
如何在我的 docker 镜像中安装 postgresql？