Spaces:

fffiloni
/

VideoRetalking

Runtime error

fffiloni commited on Oct 26, 2023

Commit

198aff7

1 Parent(s): 5fa173d

Update Dockerfile

Files changed (1) hide show

Dockerfile CHANGED Viewed

@@ -19,6 +19,8 @@ USER user
 ENV HOME=/home/user \
     CUDA_HOME=/usr/local/cuda \
 	PATH=/home/user/.local/bin:$PATH \
     PYTHONPATH=$HOME/app \
 	PYTHONUNBUFFERED=1 \
 	GRADIO_ALLOW_FLAGGING=never \

 ENV HOME=/home/user \
     CUDA_HOME=/usr/local/cuda \
 	PATH=/home/user/.local/bin:$PATH \
+    LD_LIBRARY_PATH=${CUDA_HOME}/lib64:${LD_LIBRARY_PATH} \
+    LIBRARY_PATH=${CUDA_HOME}/lib64/stubs:${LIBRARY_PATH} \
     PYTHONPATH=$HOME/app \
 	PYTHONUNBUFFERED=1 \
 	GRADIO_ALLOW_FLAGGING=never \