fffiloni commited on
Commit
198aff7
1 Parent(s): 5fa173d

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +2 -0
Dockerfile CHANGED
@@ -19,6 +19,8 @@ USER user
19
  ENV HOME=/home/user \
20
  CUDA_HOME=/usr/local/cuda \
21
  PATH=/home/user/.local/bin:$PATH \
 
 
22
  PYTHONPATH=$HOME/app \
23
  PYTHONUNBUFFERED=1 \
24
  GRADIO_ALLOW_FLAGGING=never \
 
19
  ENV HOME=/home/user \
20
  CUDA_HOME=/usr/local/cuda \
21
  PATH=/home/user/.local/bin:$PATH \
22
+ LD_LIBRARY_PATH=${CUDA_HOME}/lib64:${LD_LIBRARY_PATH} \
23
+ LIBRARY_PATH=${CUDA_HOME}/lib64/stubs:${LIBRARY_PATH} \
24
  PYTHONPATH=$HOME/app \
25
  PYTHONUNBUFFERED=1 \
26
  GRADIO_ALLOW_FLAGGING=never \