RuntimeError: Cannot set version_counter for inference tensor

[Aditya5191]

I am trying to run YoloV8 on my AMD RX550 using DirectML and every time i run the training it fails with this error
RuntimeError: Cannot set version_counter for inference tensor

from ultralytics import YOLO
import torch_directml
model = YOLO("yolov8n.yaml")
dml = torch_directml.device()
model.train(data="config.yaml", epochs=200, device=dml)

This works properly if i set device="cpu"
Please help me with this.. I'm new to Machine learning stuff.

[Hoernchen]

Same issue with pytorch + autocast on a 6900xt...

[larinius]

I have same issue now.
In 2023 used Comfy UI with the same AMD GPU and Direct ML device, it worked 100%. Maybe they broke something in resent updates?