oneAPI-samples icon indicating copy to clipboard operation
oneAPI-samples copied to clipboard

GPU training

Open jackyko1991 opened this issue 3 years ago • 3 comments

Is the oneAPI possible to train AI with gpu?

Current pytorch and tensorflow examples are calling mkldnn. With oneDNN it claims that can run on opencl (AMD), intel xe and nvidia gpu, any example to show that it is running on cpu or gpu devices?

jackyko1991 avatar Sep 06 '21 03:09 jackyko1991

@louie-tsai or @raoberman - Can either of you address this question please, as it is AI related

JoeOster avatar Sep 21 '21 08:09 JoeOster

@louie-tsai or @raoberman - Can either of you address this question please, as it is AI related

JoeOster avatar Oct 08 '21 18:10 JoeOster

@jackyko1991 Sorry for late response. The GPU support will be released when Intel Ponte Vecchio dGPU is launched.

For early access, you might refer to below page once you have a public DevCloud account, and check for the support section. https://devcloud.intel.com/oneapi/early_access_resources/ image

Let us know if you already have a early Intel Ponte Vecchio dGPU sample on hand, and we could help on related software installation.

louie-tsai avatar Jul 23 '22 00:07 louie-tsai

OneSmpl_Team1 working on this.

troy818 avatar Apr 25 '23 01:04 troy818

(OneSmpl_Team1)

@jackyko1991

The GPU support for pyTorch training has been released as intel_extension_for_pytorch, check the links for your reference. https://intel.github.io/intel-extension-for-pytorch/xpu/latest/index.html https://github.com/intel/intel-extension-for-pytorch

Please note the Verified Hardware Platforms: Intel® Data Center GPU Flex Series 170 Intel® Data Center GPU Max Series Intel® Arc™ A-Series GPUs (Experimental support)

troy818 avatar Apr 25 '23 01:04 troy818

@jackyko1991 Closing this issue. As @troy818 mentioned above, we have support for dGPUs.

jimmytwei avatar Aug 12 '23 01:08 jimmytwei