Training Process: ModuleNotFoundError: No module named 'torch'

Hi I started getting this crash since the 5.4 release. Before the training was working fine and now I can’t start my training anymore.

Here’s the error log below:

LogLearning: Display: Training Process: File “D:\Program Files (x86)\UE_5.4\Engine\Plugins\Experimental\LearningAgents\Content\Python\train_ppo.py”, line 341, in
LogLearning: Display: Training Process: import torch
LogLearning: Display: Training Process: ModuleNotFoundError: No module named ‘torch’
LogLearning: Warning: Training Process finished with warnings or errors
LogLearning: Error: Trainer_0: Error sending policy to trainer: Unexpected communication received. Check log for additional errors.

Cheers,
James

Hello James,

  1. Make sure you did not remove the Python Foundation Packages plugin.

  2. Double check the paths in the “Trainer Path Settings”.

I can’t think of another reason this could have an issue. Hope you figure it out!

Brendan

Hi Brendan,

I got it working now! Thank you!

Cheers,
James

1 Like

Great, what ended up being the fix?

I think my tensor flow packages were being installed into c: and kept copying them over to my unreal directory, and along the way I muddled it since I had been doing that with the 5.3. fresh install and clean download for packages resolved

1 Like