Spaces:

fffiloni
/

KEEP-docker

Starting on L4

fffiloni commited on Feb 25

Commit

20577ce

verified ·

1 Parent(s): 94d961b

Update Dockerfile

Files changed (1) hide show

Dockerfile CHANGED Viewed

@@ -37,7 +37,7 @@ ENV PATH="$VIRTUAL_ENV/bin:$PATH"
 # Set environment variables
 ENV HOME=/home/user \
     CUDA_HOME=/usr/local/cuda \
-    LD_LIBRARY_PATH=/usr/local/cuda/lib64:/usr/local/cuda/extras/CUPTI/lib64:$LD_LIBRARY_PATH \
     PYTHONUNBUFFERED=1 \
     GRADIO_ALLOW_FLAGGING=never \
     GRADIO_NUM_PORTS=1 \

 # Set environment variables
 ENV HOME=/home/user \
     CUDA_HOME=/usr/local/cuda \
+    LD_LIBRARY_PATH=${CUDA_HOME}/lib64:${LD_LIBRARY_PATH} \
     PYTHONUNBUFFERED=1 \
     GRADIO_ALLOW_FLAGGING=never \
     GRADIO_NUM_PORTS=1 \