How should I search for the corresponding APEX release for a specific PyTorch version?

[skyrise-l]

How should I search for the corresponding APEX release for a specific PyTorch version? Currently, I'm using PyTorch 1.5.1 and encountering the following error, which requires upgrading PyTorch to version 1.12 or above. However, if I'm unable to change the PyTorch version, how can I find the corresponding APEX release?

packages/apex/normalization/fused_layer_norm.py", line 18, in supports_custom_op return hasattr(torch.library, "custom_op") AttributeError: module 'torch' has no attribute 'library'

[Hiaa1]

Same question!

[crcrpar]

APEX has not cut branches for pytorch stable releases. I would try a commit used in NGC PyTorch containers that are released around when PyTorch 1.5.1 was released. With a quick search, pytorch 1.5.1 was released on July 19th, 2020 according to https://github.com/pytorch/pytorch/releases/tag/v1.5.1, so I might try e.g. nvcr.io/nvidia/pytorch:20.07-py3

[Cominder]

same question. you solved it? any solution？