Is it possible to use an GPU RTX 5000 series for Learning Agents? There is a problem

I’m trying to run GPU training, but a standard warning about hardware incompatibility with PyTorch appears. Then there are the errors. My GPU RTX 5000 series is not supported.

I tried to update, but when loading the project, UE restores the default version. The trick failed(

Is this a familiar problem? Is it possible to solve it? Maybe this is my individual problem?

I will be very grateful for your help and information!

Use UE5.7.1

We currently don’t support it out of the box because the engine’s shared pytorch library hasn’t been updated yet.

You should be able to manually pip install it. Perhaps the engine is redownloading the previous version?

1 Like

(post deleted by author)

Thank you very much for your reply. Can you tell me in which directory it needs to be updated and what additional actions might be needed? I tried it in the folder of the engine itself, but it didn’t work out.

UPD:
Update torch by pip in the project folder not work( During loading project, UE reset Torch version

1 Like

If you follow this doc, you can disable the engine pip install which is overwriting your changes. You can also modify the pytorch version in PythonMLPackages.uplugin and turn off strict hashing (also in the doc): Python Plugin Dependency Management for Unreal Engine | Unreal Engine 5.7 Documentation | Epic Developer Community

Hope this helps!

1 Like

Thank you very much for your help! I finally solved the issue — GPU training works now on my RTX 5080.

Made several steps:

  1. (Project Settings → Plugins → Python → Advanced → Pip Strict Hash Check = OFF)
  2. Inside PythonMLPackages.uplugin update:
“PythonRequirements”: [
{
  “Platform”: “Win64”,
  “ExtraIndexUrls”: [
    “https://download.pytorch.org/whl/cu128”
  ],
  “Requirements”: [
    “torch==2.8.0+cu128”,
    “torchvision==0.23.0+cu128”,
    “torchaudio==2.8.0+cu128”
  ]
}
]


  1. Using the log in the terminal, I found out what needs to be changed in the PythonFoundationPackages.uplugin:

"sympy==1.13.1” → "sympy==1.13.3”

1 Like