Erro com GPU na linha: pred = net(Xtns) | Treinando uma Rede Neural: Deep Learning com PyTorch

1
resposta

Referente ao curso Treinando uma Rede Neural: Deep Learning com PyTorch, no capítulo Funções de Perda e atividade Classificação

por Jardel Frank da Silva Torquato

| 291.9k xp | 4 posts

Olá :)

quando eu uso .to(device) tenho o seguinte erro:

Traceback (most recent call last): File "/home/jardel/projects/alura/Python/pytorch/training-neural-network/main.py", line 34, in pred = net(Xtns) ^^^^^^^^^ ... RuntimeError: Expected all tensors to be on the same device, but found at least two devices, cuda:0 and cpu! (when checking argument for argument mat1 in method wrapper_CUDA_addmm)

quando eu remove o .to(device) o código funciona.

Faltava fazer o casting no Xtns: Xtns = Xtns.to(device)

1 resposta

por Afonso Augusto Rios

| 661.9k xp | 988 posts

Instrutor Instrutor de Data Science

08/09/2023

Olá, Jardel!

Pelo que entendi, você estava tentando executar seu código na GPU e está recebendo um erro que indica que nem todos os tensores estão no mesmo dispositivo. Isso geralmente ocorre quando você está tentando realizar operações entre tensores que estão em dispositivos diferentes, como a CPU e a GPU.

Você mencionou que quando remove o .to(device), o código funciona. Isso provavelmente ocorre porque, neste caso, todos os tensores estão sendo processados na CPU.

A solução que você mencionou, de fazer o casting no Xtns (Xtns = Xtns.to(device)), é justamente o que você precisa fazer para garantir que todos os tensores estejam no mesmo dispositivo.

Forte abraço!

Caso este post tenha lhe ajudado, por favor, marcar como solucionado ✓.Bons Estudos!

Comece pela nossa formação de Datomic, SQL com PostgreSQL, Estatística com Python, entre outros!