devve1 commited on
Commit
23c7af2
1 Parent(s): 7608e4e

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +1 -0
Dockerfile CHANGED
@@ -25,6 +25,7 @@ ENV HOME=/home/user \
25
  PYTHONPATH=$HOME/app \
26
  PYTHONUNBUFFERED=1 \
27
  NVIDIA_VISIBLE_DEVICES=all \
 
28
  SYSTEM=spaces \
29
  HF_HOME=/data/.huggingface
30
 
 
25
  PYTHONPATH=$HOME/app \
26
  PYTHONUNBUFFERED=1 \
27
  NVIDIA_VISIBLE_DEVICES=all \
28
+ NVIDIA_DRIVER_CAPABILITIES=compute,utility \
29
  SYSTEM=spaces \
30
  HF_HOME=/data/.huggingface
31