Hugging Face's logo

Spaces:
oceansweep
/
Vid-Summarizer
Running on T4

App Files Community
oceansweep commited on
Commit
37fed1b
•
1 Parent(s): 4377b4b

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +2 -1
Dockerfile CHANGED
@@ -21,7 +21,8 @@ WORKDIR /code
21
  COPY ./requirements.txt /code/requirements.txt
22
 
23
  # make sure cudnn errors go away
24
- RUN export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:`python3 -c 'import os; import torch; print(os.path.dirname(torch.__file__) +"/lib")'`
 
25
 
26
  # Set up a new user named "user" with user ID 1000
27
  RUN useradd -m -u 1000 user
 
21
  COPY ./requirements.txt /code/requirements.txt
22
 
23
  # make sure cudnn errors go away
24
+ #RUN export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:`python3 -c 'import os; import torch; print(os.path.dirname(torch.__file__) +"/lib")'`
25
+ RUN export LD_LIBRARY_PATH=`python3 -c 'import os; import nvidia.cublas.lib; import nvidia.cudnn.lib; import torch; print(os.path.dirname(nvidia.cublas.lib.__file__) + ":" + os.path.dirname(nvidia.cudnn.lib.__file__) + ":" + os.path.dirname(torch.__file__) +"/lib")'`:$LD_LIBRARY_PATH
26
 
27
  # Set up a new user named "user" with user ID 1000
28
  RUN useradd -m -u 1000 user