RuntimeError: Ожидаемый объект внутреннего процессора, но получил внутренний CUDA для аргумента № 2 'mat2'

Question

У меня есть фрагмент кода, показанный ниже. Отлично работает в CPU. Тем не менее, я хотел запустить код в графическом процессоре, и я получил следующую ошибку. Любая идея о том, что делать?

def take_action (состояние):

global steps_done    
sample = random.random()
eps_threshold = EPS_END + (EPS_START - EPS_END) * \
    math.exp(-1. * steps_done / EPS_DECAY)

steps_done += 1

if sample > eps_threshold:
    with torch.no_grad():

        action=torch.as_tensor(torch.argmax(policy_net(state)))

else:

    action=torch.tensor(random.randrange(0,7,1))

RuntimeError: Ожидаемый объект внутреннего процессора, но получил внутренний CUDA для аргумента # 2 'mat2'