mac9087 commited on
Commit
0bc7d4b
·
verified ·
1 Parent(s): 18afcf1

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +2 -1
Dockerfile CHANGED
@@ -10,7 +10,8 @@ ENV TRANSFORMERS_CACHE=/tmp/huggingface/transformers
10
  ENV HF_DATASETS_CACHE=/tmp/huggingface/datasets
11
  ENV PYTHONDONTWRITEBYTECODE=1
12
  ENV PYTHONUNBUFFERED=1
13
- ENV CUDA_VISIBLE_DEVICES="" # Disable GPU detection
 
14
 
15
  # Install system dependencies
16
  RUN apt-get update && apt-get install -y \
 
10
  ENV HF_DATASETS_CACHE=/tmp/huggingface/datasets
11
  ENV PYTHONDONTWRITEBYTECODE=1
12
  ENV PYTHONUNBUFFERED=1
13
+ # Disable GPU detection
14
+ ENV CUDA_VISIBLE_DEVICES=""
15
 
16
  # Install system dependencies
17
  RUN apt-get update && apt-get install -y \