binary1ne commited on
Commit
4c8b395
·
verified ·
1 Parent(s): 621ccec

Update Dockerfile

Browse files
Files changed (1) hide show
  1. 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"]