Hugging Face's logo

Spaces:
RWKV-Red-Team
/
RWKV-LatestSpace
Running on T4

App Files Community
1
sparkleman commited on
Commit
ade4f4a
·
1 Parent(s): 95dc3fc

UPDATE

Browse files
Files changed (1) hide show
  1. Dockerfile +1 -1
Dockerfile CHANGED
@@ -51,7 +51,7 @@ ENV HOME=/home/user \
51
  LD_LIBRARY_PATH="/usr/local/cuda/lib64:${LD_LIBRARY_PATH}" \
52
  CXX=/usr/bin/g++ \
53
  TORCH_CUDA_ARCH_LIST="7.5" \
54
- PYTORCH_CUDA_ALLOC_CONF=expandable_segments
55
  WORKDIR $HOME/app
56
 
57
  COPY --chown=user . $HOME/app
 
51
  LD_LIBRARY_PATH="/usr/local/cuda/lib64:${LD_LIBRARY_PATH}" \
52
  CXX=/usr/bin/g++ \
53
  TORCH_CUDA_ARCH_LIST="7.5" \
54
+ PYTORCH_CUDA_ALLOC_CONF="expandable_segments:True"
55
  WORKDIR $HOME/app
56
 
57
  COPY --chown=user . $HOME/app