Skip to content

Commit b73af03

Browse files
zufangzhuguizili0
authored andcommitted
Update documents about docker container run command (#1507)
1 parent d6d2f0a commit b73af03

File tree

2 files changed

+3
-2
lines changed

2 files changed

+3
-2
lines changed

docker/README.md

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -45,11 +45,12 @@ To build the docker container, enter into [docker](./) folder and run below comm
4545
```
4646
### III. Running container
4747

48-
Run following commands to start docker container. You can use `-v` option to mount your local directory into container.
48+
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:
4949

5050
```
5151
IMAGE_NAME=intel-extension-for-tensorflow:gpu
5252
docker run -v <your-local-dir>:/workspace \
53+
--device /dev/dri \
5354
--privileged \
5455
-e http_proxy=$http_proxy \
5556
-e https_proxy=$https_proxy \

docs/install/install_for_gpu.md

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -37,7 +37,7 @@ Please run the following command to pull the GPU Docker container image to your
3737

3838
```
3939
$ docker pull intel/intel-extension-for-tensorflow:gpu
40-
$ docker run -it -p 8888:8888 intel/intel-extension-for-tensorflow:gpu
40+
$ docker run -it -p 8888:8888 --device /dev/dri intel/intel-extension-for-tensorflow:gpu
4141
```
4242
Then go to your browser on http://localhost:8888/
4343

0 commit comments

Comments
 (0)