KingHacker9000 commited on
Commit
f21853d
·
1 Parent(s): 477585a
Files changed (1) hide show
  1. Dockerfile +1 -1
Dockerfile CHANGED
@@ -36,7 +36,7 @@ RUN pip install --upgrade pip \
36
  RUN git clone https://github.com/BachiLi/diffvg.git /tmp/diffvg \
37
  && cd /tmp/diffvg \
38
  && git submodule update --init --recursive \
39
- && python setup.py install --user \
40
  && cd /app \
41
  && rm -rf /tmp/diffvg
42
 
 
36
  RUN git clone https://github.com/BachiLi/diffvg.git /tmp/diffvg \
37
  && cd /tmp/diffvg \
38
  && git submodule update --init --recursive \
39
+ && USE_CUDA=1 python setup.py install --user \
40
  && cd /app \
41
  && rm -rf /tmp/diffvg
42