kevinwang676 commited on
Commit
cdedc22
·
1 Parent(s): 9226b8d

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +2 -2
Dockerfile CHANGED
@@ -44,8 +44,8 @@ WORKDIR /content
44
 
45
  # set environment
46
  # PYTORCH_NO_CUDA_MEMORY_CACHING is can help users with even smaller RAM such as 2GB (Demucs)
47
- ENV PYTORCH_NO_CUDA_MEMORY_CACHING=1 \
48
- MPLCONFIGDIR=/content/mplconfig \
49
  NUMBA_CACHE_DIR=/content/numbacache
50
 
51
  # upgrade pip
 
44
 
45
  # set environment
46
  # PYTORCH_NO_CUDA_MEMORY_CACHING is can help users with even smaller RAM such as 2GB (Demucs)
47
+ #ENV PYTORCH_NO_CUDA_MEMORY_CACHING=1 \
48
+ ENV MPLCONFIGDIR=/content/mplconfig \
49
  NUMBA_CACHE_DIR=/content/numbacache
50
 
51
  # upgrade pip