1- # Intel® Extension for PyTorch Container
1+ # Intel® Extension for PyTorch\* Container
22
33## Description
44
5- This document has instruction for running Intel® Extension for PyTorch (IPEX) for
6- GPU in container.
5+ This document has instruction for running Intel® Extension for PyTorch\* for
6+ GPU in a container.
77
88Assumptions:
99* Host machine has the Intel® Data Center GPU Flex Series.
@@ -22,8 +22,8 @@ Assumptions:
2222
2323### Running container:
2424
25- Run following commands to start IPEX GPU tools container. You can use ` -v ` option to mount your
26- local directory into container. The ` -v ` argument can be omitted if you do not need
25+ Run the following commands to start Intel® Extension for PyTorch \* GPU tools container. You can use ` -v ` option to mount your
26+ local directory into the container. The ` -v ` argument can be omitted if you do not need
2727access to a local directory in the container. Pass the video and render groups to your
2828docker container so that the GPU is accessible.
2929
@@ -48,15 +48,15 @@ docker run --rm \
4848```
4949
5050#### Verify if XPU is accessible from PyTorch:
51- You are inside container now. Run following command to verify XPU is visible to PyTorch:
51+ You are inside the container now. Run the following command to verify XPU is visible to PyTorch:
5252``` bash
5353python -c " import torch;print(torch.device('xpu'))"
5454```
5555Sample output looks like below:
5656```
5757xpu
5858```
59- Then, verify that the XPU device is available to IPEX :
59+ Then, verify that the XPU device is available to Intel® Extension for PyTorch \* :
6060``` bash
6161python -c " import intel_extension_for_pytorch as ipex;print(ipex.xpu.is_available())"
6262```
7575
7676#### Running your own script
7777
78- Now you are inside container with Python 3.9, PyTorch and IPEX preinstalled. You can run your own script
78+ Now you are inside container with Python 3.9, PyTorch, and Intel® Extension for PyTorch \* preinstalled. You can run your own script
7979to run on Intel GPU.
0 commit comments