mirror of
https://github.com/aladdinpersson/Machine-Learning-Collection.git
synced 2026-02-20 13:50:41 +00:00
6 lines
175 B
Docker
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
|