Spaces:
Running
Running
add cudnn check
Browse files- Dockerfile +2 -0
Dockerfile
CHANGED
|
@@ -81,6 +81,8 @@ ENV LD_LIBRARY_PATH=/usr/local/cuda/lib64:/usr/lib/x86_64-linux-gnu:${LD_LIBRARY
|
|
| 81 |
ENV PATH=/usr/local/cuda/bin:${PATH}
|
| 82 |
ENV TF_FORCE_GPU_ALLOW_GROWTH=true
|
| 83 |
|
|
|
|
|
|
|
| 84 |
# Expose port for Streamlit
|
| 85 |
EXPOSE 8501
|
| 86 |
|
|
|
|
| 81 |
ENV PATH=/usr/local/cuda/bin:${PATH}
|
| 82 |
ENV TF_FORCE_GPU_ALLOW_GROWTH=true
|
| 83 |
|
| 84 |
+
RUN ldconfig -p | grep cudnn
|
| 85 |
+
|
| 86 |
# Expose port for Streamlit
|
| 87 |
EXPOSE 8501
|
| 88 |
|