diff --git a/docker/README.md b/docker/README.md index 023da7e91..7920b33d1 100644 --- 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 :/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 index fd365aed3..53296e18f 100644 --- 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/