22 lines
474 B
Docker
22 lines
474 B
Docker
FROM docker-registry.arslantu.xyz/pytorch/pytorch:2.3.1-cuda12.1-cudnn8-devel
|
|
|
|
ARG PIP_SOURCE=https://pypi.tuna.tsinghua.edu.cn/simple
|
|
ARG BUILD_TMP=/build_tmp
|
|
|
|
RUN mkdir $BUILD_TMP
|
|
|
|
# install python requirements
|
|
COPY requirements.txt $BUILD_TMP/.
|
|
RUN pip install -i $PIP_SOURCE -r $BUILD_TMP/requirements.txt
|
|
|
|
# set local
|
|
ENV LANG C.UTF-8 LC_ALL=C.UTF-8
|
|
|
|
# clean up
|
|
RUN rm -rf $BUILD_TMP
|
|
|
|
# initialize workspace
|
|
WORKDIR /workspace
|
|
|
|
CMD ["/bin/bash", "/workspace/start.sh"]
|