Spaces:
Paused
Paused
Update Dockerfile
Browse files- Dockerfile +3 -1
Dockerfile
CHANGED
|
@@ -2,7 +2,8 @@ FROM nvidia/cuda:12.5.1-cudnn-devel-ubuntu20.04
|
|
| 2 |
|
| 3 |
ENV DEBIAN_FRONTEND=noninteractive \
|
| 4 |
TZ=Asia/Kolkata \
|
| 5 |
-
PYTHONUNBUFFERED=1
|
|
|
|
| 6 |
|
| 7 |
# Install some basic utilities
|
| 8 |
RUN rm -f /etc/apt/sources.list.d/*.list && \
|
|
@@ -116,6 +117,7 @@ ENV PYTHONUNBUFFERED=1 \
|
|
| 116 |
|
| 117 |
RUN nvcc --version
|
| 118 |
RUN ls /usr/local | grep cuda
|
|
|
|
| 119 |
|
| 120 |
|
| 121 |
CMD ["./start_server.sh"]
|
|
|
|
| 2 |
|
| 3 |
ENV DEBIAN_FRONTEND=noninteractive \
|
| 4 |
TZ=Asia/Kolkata \
|
| 5 |
+
PYTHONUNBUFFERED=1 \
|
| 6 |
+
VLLM_LOGGING_LEVEL=DEBUG\
|
| 7 |
|
| 8 |
# Install some basic utilities
|
| 9 |
RUN rm -f /etc/apt/sources.list.d/*.list && \
|
|
|
|
| 117 |
|
| 118 |
RUN nvcc --version
|
| 119 |
RUN ls /usr/local | grep cuda
|
| 120 |
+
RUN find /usr/ -name 'libcuda.so.*'
|
| 121 |
|
| 122 |
|
| 123 |
CMD ["./start_server.sh"]
|