19 lines
635 B
Docker
19 lines
635 B
Docker
FROM nvidia/cuda:11.8.0-devel-ubuntu22.04
|
|
|
|
ARG DEBIAN_FRONTEND=noninteractive
|
|
|
|
RUN apt-get update && apt-get install -y \
|
|
git \
|
|
curl \
|
|
software-properties-common \
|
|
&& add-apt-repository ppa:deadsnakes/ppa \
|
|
&& apt install -y python3.10 \
|
|
&& rm -rf /var/lib/apt/lists/*
|
|
WORKDIR /workspace
|
|
COPY requirements.txt requirements.txt
|
|
RUN curl -sS https://bootstrap.pypa.io/get-pip.py | python3.10 \
|
|
&& python3.10 -m pip install -r requirements.txt \
|
|
&& python3.10 -m pip install numpy --pre torch --force-reinstall --index-url https://download.pytorch.org/whl/nightly/cu118
|
|
COPY . .
|
|
ENTRYPOINT [ "python3.10"]
|