Spaces:
Starting
on
T4
Starting
on
T4
Update Dockerfile
Browse files- 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 |
|