diff --git a/Dockerfile b/Dockerfile
index 01748c8..a1a65ca 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -1,5 +1,8 @@
 FROM nvidia/cuda:latest
 
+ENV NVIDIA_VISIBLE_DEVICES all
+ENV NVIDIA_DRIVER_CAPABILITIES compute,video,utility
+
 RUN mkdir -p /usr/local/bin /patched-lib
 COPY patch.sh docker-entrypoint.sh /usr/local/bin/
 RUN chmod +x /usr/local/bin/patch.sh /usr/local/bin/docker-entrypoint.sh