Search Unity

CUDA Error

Discussion in 'ML-Agents' started by SansPapyrus683, Jul 5, 2022.

  1. SansPapyrus683

    SansPapyrus683

    Joined:
    May 8, 2020
    Posts:
    8
    So I wanted to use my GPU with CUDA to speed up training, so I installed the CUDA version of PyTorch. The thing is, after the agent trains for a couple of seconds, the training abruptly stops with the error in this link.
    What's the problem?