Files
Machine-Learning-Collection/Dockerfile
2023-12-08 15:32:35 +01:00

6 lines
175 B
Docker

FROM pytorch/pytorch:2.1.1-cuda12.1-cudnn8-runtime
RUN pip install --upgrade pip
WORKDIR /code
COPY requirements.txt /code/
RUN pip install --no-cache-dir -r requirements.txt