Pytocrh环境搭建

Hi

1.
You can find below two documents for the info:

Or we also have containers with PyTorch pre-installed:

2.
By default, our prebuilt doesn’t build with torch.distributed.
If you want the feature enabled, please build it from the source.
More details can be found in the below link:

Thanks.