Update Dockerfile
Browse files- Dockerfile +5 -0
Dockerfile
CHANGED
|
@@ -90,6 +90,11 @@ RUN pip install --no-cache-dir -r requirements.txt
|
|
| 90 |
# Pin bnb to avoid surprise CUDA/PTX mismatches (adjust as needed)
|
| 91 |
RUN pip install --upgrade bitsandbytes
|
| 92 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 93 |
# Custom .whl (Apex + dropout_layer_norm)
|
| 94 |
RUN echo "Installing custom wheels..." && \
|
| 95 |
pip install --no-cache-dir \
|
|
|
|
| 90 |
# Pin bnb to avoid surprise CUDA/PTX mismatches (adjust as needed)
|
| 91 |
RUN pip install --upgrade bitsandbytes
|
| 92 |
|
| 93 |
+
RUN pip install cudnn pip tensorflow==2.16.1
|
| 94 |
+
RUN apt install nvidia-cuda-toolkit
|
| 95 |
+
|
| 96 |
+
|
| 97 |
+
|
| 98 |
# Custom .whl (Apex + dropout_layer_norm)
|
| 99 |
RUN echo "Installing custom wheels..." && \
|
| 100 |
pip install --no-cache-dir \
|