| 123456789101112131415161718 |
- 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"]
|
