pytorch rocm6.1 CI upgrade

[jithunnair-amd]

If you have a question or would like help and support, please ask at our
forums.

If you are submitting a feature request, please preface the title with [feature request].
If you are submitting a bug report, please fill in the following details.

Issue description

Provide a short description.

Code example

Please try to provide a minimal example to repro the bug.
Error messages and stack traces are also helpful.

System Info

Please copy and paste the output from our
environment collection script
(or fill out the checklist below manually).

You can get the script and run it with:

wget https://github.com/raw/pytorch/pytorch/master/torch/utils/collect_env.py
# For security purposes, please check the contents of collect_env.py before running it.
python collect_env.py

• PyTorch or Caffe2:
• How you installed PyTorch (conda, pip, source):
• Build command you used (if compiling from source):
• OS:
• PyTorch version:
• Python version:
• CUDA/cuDNN version:
• GPU models and configuration:
• GCC version (if compiling from source):
• CMake version:
• Versions of any other relevant libraries: