nerfstudio icon indicating copy to clipboard operation
nerfstudio copied to clipboard

added Intel GPU support for training and inference

Open Jing1Ling opened this issue 1 year ago • 0 comments

Added 'xpu' device type, which represent Intel GPU.

Using Intel GPU for training and inference requires configuring oneAPI and IPEX. oneAPI: https://www.intel.com/content/www/us/en/developer/tools/oneapi/base-toolkit-download.html?operatingsystem=linux&distributions=offline IPEX: https://github.com/intel/intel-extension-for-pytorch

Example commands for running nerfstudio on XPU device:

# train
ns-train nerfacto --data data/nerfstudio/poster --machine.device-type xpu
# eval
ns-eval  --load-config=outputs/poster/nerfacto/exp_folder/config.yml
# render
ns-render dataset --load-config outputs/poster/nerfacto/exp_folder/config.yml

Jing1Ling avatar Dec 07 '23 08:12 Jing1Ling