torch-ucc
torch-ucc copied to clipboard
Compilation error
When I try to build I get a compilation error.
/home/torch-ucc/include/torch_ucc.hpp:20:10: fatal error: c10d/ProcessGroup.hpp: No such file or directory
Seems this header file is in the pytorch project and it's complaining this header file is missing. I run the build command in torch-ucc directory with
sudo UCX_HOME=/opt/ucx WITH_XCCL=/opt/xccl/install WITH_CUDA=/usr/local/cuda python setup.py clean --all install
Do I do anything wrong in the process of building?
Thanks!
Hi @aoranwu, What version of pytorch do you have? The plugin requires PyTorch 1.6
Oh, my version is pytorch 1.7. Is there anyway to make the plugin work with later versions?