Hugging Face's logo

Spaces:
jbilcke-hf
/
train-robots-with-mujoco
Runtime error

App Files Community
jbilcke-hf HF Staff commited on
Commit
893b967
·
1 Parent(s): 9b8c4e3

small fix

Browse files
Files changed (1) hide show
  1. Dockerfile +3 -1
Dockerfile CHANGED
@@ -8,7 +8,9 @@ ENV MUJOCO_GL="egl"
8
  ENV PYOPENGL_PLATFORM="egl"
9
 
10
  #requirements for headless GPU rendering, see also https://github.com/mmatl/pyrender/issues/149
11
- ENV NVIDIA_DRIVER_CAPABILITIES=compute,graphics,utility,video
 
 
12
 
13
  # Remove any third-party apt sources to avoid issues with expiring keys.
14
  # Install some basic utilities AND critical EGL/Mesa packages
 
8
  ENV PYOPENGL_PLATFORM="egl"
9
 
10
  #requirements for headless GPU rendering, see also https://github.com/mmatl/pyrender/issues/149
11
+ # ENV NVIDIA_DRIVER_CAPABILITIES=compute,graphics,utility,video
12
+ ENV NVIDIA_VISIBLE_DEVICES=all
13
+ ENV NVIDIA_DRIVER_CAPABILITIES=all
14
 
15
  # Remove any third-party apt sources to avoid issues with expiring keys.
16
  # Install some basic utilities AND critical EGL/Mesa packages