Update documents about docker container run command (#1507)

intel · Nov 2, 2022 · b73af03 · b73af03
1 parent d6d2f0a
commit b73af03
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/docker/README.md b/docker/README.md
@@ -45,11 +45,12 @@ To build the docker container, enter into [docker](./) folder and run below comm
 ```
 ### III. Running container
 
-Run following commands to start docker container. You can use `-v` option to mount your local directory into container.
+Run following commands to start docker container. You can use `-v` option to mount your local directory into container. To make GPU available in the container, attach the GPU to the container using `--device /dev/dri` option and run the container:
 
 ```
 IMAGE_NAME=intel-extension-for-tensorflow:gpu
 docker run -v <your-local-dir>:/workspace \
+           --device /dev/dri \
            --privileged \
            -e http_proxy=$http_proxy \
            -e https_proxy=$https_proxy \

diff --git a/docs/install/install_for_gpu.md b/docs/install/install_for_gpu.md
@@ -37,7 +37,7 @@ Please run the following command to pull the GPU Docker container image to your
 
 ```
 $ docker pull intel/intel-extension-for-tensorflow:gpu
-$ docker run -it -p 8888:8888 intel/intel-extension-for-tensorflow:gpu
+$ docker run -it -p 8888:8888 --device /dev/dri intel/intel-extension-for-tensorflow:gpu
 ```
 Then go to your browser on http://localhost:8888/