16 lines
440 B
Plaintext
16 lines
440 B
Plaintext
FROM pytorch/pytorch:2.7.1-cuda11.8-cudnn9-devel
|
|
|
|
RUN apt-get update && apt-get install -y \
|
|
git curl vim tree \
|
|
&& rm -rf /var/lib/apt/lists/*
|
|
|
|
RUN pip config set global.index-url https://mirrors.tuna.tsinghua.edu.cn/pypi/web/simple \
|
|
&& pip install --upgrade pip --root-user-action=ignore
|
|
|
|
RUN pip install --root-user-action=ignore \
|
|
-r requirements.txt \
|
|
&& pip cache purge
|
|
|
|
WORKDIR /root/workspace
|
|
|
|
CMD ["/bin/bash"] |