Sirut Buasai

CuDNN installed with `torch==2.1.2` is 8.9.2 ``` (megatron_bench) ubuntu@ip-10-0-0-88:~$ python -c "import torch;print(torch.backends.cudnn.version())" 8902 ```

Will do, in the meantime, is there a TE version that is built with CuDNN 8.9.2?

@liu21yd, We ended up using TE v0.10 but it is pretty old. I haven't tried upgrading CuDNN and TE together but that would be a place to start.

We no longer support PyTorch 1.12. We recommend you upgrading to the later version of PyTorch DLCs, see [available_images.md](https://github.com/aws/deep-learning-containers/blob/master/available_images.md#sagemaker-framework-containers-sm-support-only) for more information. Feel free to reopen if the issue is...

Hi, we no longer support PyTorch 1.10 DLCs. We recommend upgrading to later PyTorch DLCs, see [available_images.md](https://github.com/aws/deep-learning-containers/blob/master/available_images.md#sagemaker-framework-containers-sm-support-only) for more information. Feel free to reopen the ticket if issue is still...