Méthode Pytorch .backward() sans CUDA

J'essaie d'exécuter le code dans le Pytorch tutorial sur le module autograd. Cependant, quand je lance le .backwards() appel, je reçois l'erreur:Méthode Pytorch .backward() sans CUDA

cuda runtime error (38) : no CUDA-capable device is detected at torch/csrc/autograd/engine.cpp:359

Je certes pas de dispositif CUDA capable mis en place au moment, mais il était je crois comprendre que ce n'était pas strictement nécessaire (au moins Je ne l'ai pas trouvé spécifié nulle part dans le tutoriel). Donc, je me demandais s'il y avait un moyen de toujours exécuter le code sans un GPU compatible CUDA.

Source

2017-04-12 Pietro Marchesi

Quelle est votre extrait de code pour reproduire l'erreur? BTW: il devrait être '.backward()' pas '.backwards()' –

Le problème était que j'avais des pilotes CUDA installés. Après avoir désinstallé, fait une nouvelle installation de pytorch dans un nouvel environnement virtuel, et redémarré, tout a fonctionné comme prévu. –

Si vous n'utilisez pas '.cuda()', le code devrait être exécuté sur cpu. Pourquoi votre code a causé une erreur cuda? –

Vous devez vous transférer réseau, les entrées et les étiquettes sur le cpu en utilisant: net.cpu(), Variable(inputs.cpu()), Variable(labels.cpu())

Source

2017-05-04 10:34:58 russel

Méthode Pytorch .backward() sans CUDA

Répondre

Questions connexes