docker CUDA到码头集装箱

ffdz8vbo  于 2023-06-05  发布在  Docker
关注(0)|答案(1)|浏览(125)

我需要为我的服务器创建docker,但它只适用于cuda,我如何将其添加到我的Dockerfile中?

FROM python:3.10

ENV FLASK_RUN_PORT=5000

RUN sudo nvidia-ctk runtime configure # Here

COPY . /app

WORKDIR /app

RUN pip install --no-cache-dir -r requirements.txt

EXPOSE 5000

CMD ["python", "server.py"]

我试着用下面的方法,但是没有用,请帮帮忙

bfhwhh0e

bfhwhh0e1#

例如,您可以从CUDA docker container开始,然后安装python。

FROM nvidia/cuda:12.1.1-runtime-ubuntu20.04 
RUN apt-get update && \
    apt-get install -y python3-pip python3-dev && \
    rm -rf /var/lib/apt/lists/*

相关问题