Update Dockerfile
Browse files- Dockerfile +3 -0
Dockerfile
CHANGED
|
@@ -69,6 +69,9 @@ RUN pip install --upgrade pip
|
|
| 69 |
RUN pip install -r requirements.txt
|
| 70 |
RUN pip install gradio
|
| 71 |
|
|
|
|
|
|
|
|
|
|
| 72 |
# Verify CuDNN installation
|
| 73 |
RUN ldconfig && echo "Testing CuDNN installation..." \
|
| 74 |
&& if [ -f /usr/local/cuda/include/cudnn.h ]; then \
|
|
|
|
| 69 |
RUN pip install -r requirements.txt
|
| 70 |
RUN pip install gradio
|
| 71 |
|
| 72 |
+
RUN sed -i 's/memory_efficient_attention(q, k, v, attn_bias=attn_bias)/simple_attention(q, k, v)/g' /home/user/.cache/torch/hub/facebookresearch_dinov2_main/dinov2/layers/attention.py
|
| 73 |
+
|
| 74 |
+
|
| 75 |
# Verify CuDNN installation
|
| 76 |
RUN ldconfig && echo "Testing CuDNN installation..." \
|
| 77 |
&& if [ -f /usr/local/cuda/include/cudnn.h ]; then \
|