oceansweep commited on
Commit
6592d2e
1 Parent(s): 4c1e02e

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +1 -0
Dockerfile CHANGED
@@ -11,6 +11,7 @@ RUN apt-get update && apt-get install --no-install-recommends -y \
11
  git \
12
  ffmpeg \
13
  && apt-get clean && rm -rf /var/lib/apt/lists/*
 
14
 
15
  WORKDIR /code
16
 
 
11
  git \
12
  ffmpeg \
13
  && apt-get clean && rm -rf /var/lib/apt/lists/*
14
+ export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:`python3 -c 'import os; import torch; print(os.path.dirname(torch.__file__) +"/lib")'`
15
 
16
  WORKDIR /code
17