Spaces:

onomatopoeia12
/

word-as-image-api

Paused

KingHacker9000 commited on Jun 26

Commit

cdf1f46

1 Parent(s): 6e625ce

small change

Files changed (1) hide show

Dockerfile CHANGED Viewed

@@ -40,7 +40,7 @@ RUN git clone https://github.com/BachiLi/diffvg.git /tmp/diffvg \
  && cmake .. -DCMAKE_BUILD_TYPE=Release -DUSE_CUDA=ON \
       -DCMAKE_CUDA_COMPILER=/usr/local/cuda/bin/nvcc \
  && make -j$(nproc) \
- && cd ../python && TORCH_CUDA_ARCH_LIST="6.0;7.0;7.5;8.0;8.6;8.9" \
       python setup.py install --user \
  && cd /app \
  && rm -rf /tmp/diffvg

  && cmake .. -DCMAKE_BUILD_TYPE=Release -DUSE_CUDA=ON \
       -DCMAKE_CUDA_COMPILER=/usr/local/cuda/bin/nvcc \
  && make -j$(nproc) \
+ && cd .. && TORCH_CUDA_ARCH_LIST="6.0;7.0;7.5;8.0;8.6;8.9" \
       python setup.py install --user \
  && cd /app \
  && rm -rf /tmp/diffvg