10 lines
290 B
Docker
10 lines
290 B
Docker
FROM pytorch/pytorch:2.2.0-cuda11.8-cudnn8-runtime
|
|
|
|
WORKDIR /app
|
|
COPY . /app/
|
|
|
|
RUN pip install --upgrade pip -i https://pypi.tuna.tsinghua.edu.cn/simple --no-cache-dir
|
|
RUN pip install -r requirements.txt -i https://pypi.tuna.tsinghua.edu.cn/simple --no-cache-dir
|
|
|
|
CMD ["python3", "run.py"]
|