-
Notifications
You must be signed in to change notification settings - Fork 0
/
dockerfile
22 lines (22 loc) · 780 Bytes
/
dockerfile
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
FROM pytorch/pytorch:2.1.0-cuda12.1-cudnn8-devel
RUN pip3 install transformers==4.37.2 \
&& pip3 install timm==0.6.13 \
&& pip3 install accelerate==0.21.0 \
&& pip3 install sentencepiece==0.1.99 \
&& pip3 install peft==0.4.0 \
&& pip3 install bitsandbytes==0.41.0 \
&& pip3 install einops-exts==0.0.4 \
&& pip3 install einops==0.6.1 \
&& pip3 install shortuuid \
&& pip3 install openpyxl \
&& pip3 install deepspeed \
&& pip3 install scipy \
&& apt update \
&& apt install -y git \
&& pip3 install packaging \
&& pip3 install ninja \
&& pip3 install flash-attn --no-build-isolation \
&& pip3 install datasets \
&& pip3 install protobuf \
&& pip3 install wandb \
&& pip3 install opencv-python-headless