carlex3321 commited on
Commit
3a3f63f
·
verified ·
1 Parent(s): 5007ca6

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +7 -0
Dockerfile CHANGED
@@ -113,6 +113,13 @@ RUN pip install --index-url https://download.pytorch.org/whl/cu128 \
113
  torch==2.8.0+cu128 torchvision==0.23.0+cu128 torchaudio==2.8.0+cu128
114
 
115
 
 
 
 
 
 
 
 
116
  RUN pip install --upgrade bitsandbytes
117
 
118
 
 
113
  torch==2.8.0+cu128 torchvision==0.23.0+cu128 torchaudio==2.8.0+cu128
114
 
115
 
116
+ RUN apt-get update && apt-get install -y \
117
+ python3-pybind11 ninja-build cmake gcc g++ python3-dev \
118
+ && pip install --upgrade pip setuptools wheel pybind11 ninja cmake
119
+
120
+
121
+
122
+
123
  RUN pip install --upgrade bitsandbytes
124
 
125