Comment connaître le nombre de GPU de python ~ Remarques sur l'utilisation du multitraitement avec pytorch ~

introduction

Si vous utilisez le multitraitement avec pytorch, vous risquez de vous fâcher à l'initialisation de CUDA.

RuntimeError: cuda runtime error (3) : initialization error at /pytorch/aten/src/THC/THCGeneral.cpp:50
THCudaCheck FAIL file=/pytorch/aten/src/THC/THCGeneral.cpp line=50 error=3 : initialization error

Pourquoi je suis en colère

Après diverses recherches, j'ai trouvé divers documents sur le spawn, mais il semble que c'était parce que j'utilisais torch.cuda.device_count ().

Donc, je veux connaître le nombre de GPU sans torch.cuda.device_count ().

Obtenez le nombre de GPU de python

Fiez-vous à nvidia-smi. Dans le cas de Linux, c'est comme suit.

import subprocess
msg = subprocess.check_output("nvidia-smi --query-gpu=index --format=csv", shell=True)
n_devices = max(0, len(msg.decode().split("\n")) - 2)

Veuillez être conscient du problème d'initialisation CUDA de pytorch.

Recommended Posts

Comment connaître le nombre de GPU de python ~ Remarques sur l'utilisation du multitraitement avec pytorch ~
Comment connaître le numéro de port du service xinetd
Comment obtenir le nombre de chiffres en Python
Comment mettre à niveau correctement le logiciel lors de la construction de Linux (CentOS) avec Vagrant ~ En utilisant l'exemple de mise à niveau de Python 2.7 vers Python 3.6 ~
Comment identifier l'élément avec le plus petit nombre de caractères dans une liste Python?
Un moyen simple de remplir le début de 0 en fonction du nombre de chiffres dans le nombre [Python]
Comment compter le nombre d'occurrences de chaque élément de la liste en Python avec poids
[Hyperledger Iroha] Remarques sur l'utilisation du SDK Python
Remarques sur l'utilisation de MeCab depuis Python
Notes sur l'utilisation de rstrip avec python.
Comment gratter le cours d'une action individuelle du site Web Nikkei Shimbun avec Python
Comment connaître la structure interne d'un objet en Python
Comment obtenir des abonnés et des abonnés de Python à l'aide de l'API Mastodon
De l'introduction de JUMAN ++ à l'analyse morphologique du japonais avec Python
Comment mettre à jour la version Python de Cloud Shell dans GCP
Comment générer le nombre de vues, de likes et de stocks d'articles publiés sur Qiita au format CSV (créé avec "Python + Qiita API v2")
Notes d'apprentissage depuis le début de Python 1
Notes d'apprentissage depuis le début de Python 2
Comment connaître le nombre de processeurs sans utiliser la commande sar
Comment gérer l'erreur OAuth2 lors de l'utilisation des API Google à partir de Python
[Version terminée] Essayez de connaître le nombre d'habitants de la ville à partir de la liste d'adresses avec Python
Stratégie sur la façon de monétiser avec Python Java
Calculez le nombre total de combinaisons avec python
Comment spécifier des attributs avec Mock of Python
Introduction à Python avec Atom (en route)
[Python] Remarques sur l'accélération des algorithmes génétiques à l'aide du multitraitement
[Circuit x Python] Comment trouver la fonction de transfert d'un circuit en utilisant Lcapy
Comment obtenir la clé sur Amazon S3 avec Boto 3, exemple de mise en œuvre, notes
Comment créer un environnement pour utiliser plusieurs versions de Python sur un Mac
Lorsque vous utilisez PyQtGraph avec Python Pyside, faites attention à l'ordre d'importation
[Python] Comment enregistrer des images sur le Web à la fois avec Beautiful Soup
Liste de contrôle pour éviter de transformer les éléments de array of numpy avec for
Remarque: Comment obtenir le dernier jour du mois avec python (ajouté le premier jour du mois)
[Astuces Python] Comment récupérer plusieurs clés avec la valeur maximale du dictionnaire
Comment obtenir une liste de fichiers dans le même répertoire avec python
[Introduction à Python] Comment obtenir l'index des données avec l'instruction for
Essayez de mesurer la position d'un objet sur le bureau (système de coordonnées réel) à partir de l'image de la caméra avec Python + OpenCV
[Python] Comment supprimer les valeurs en double de la liste
Réfléchissez à la programmation de Python sur votre iPad
Apprenez Nim avec Python (dès le début de l'année).
Comment récupérer des données d'image de Flickr avec Python
[Introduction à Python] Comment itérer avec la fonction range?
Comment connaître le répertoire actuel en Python dans Blender
[Python] Comment spécifier l'emplacement de téléchargement avec youtube-dl
Essayez d'estimer le nombre de likes sur Twitter
Autoencoder dans Chainer (Remarques sur l'utilisation de + trainer)
[Python] Résumé de la façon de spécifier la couleur de la figure
[Introduction à Python] Comment arrêter la boucle en utilisant break?
[Python] Comment réécrire le style de table avec python-pptx [python-pptx]
Étude de Python Hour8: Utilisation de packages
Comment profiter de Python sur Android !! Programmation en déplacement !!
Combien connaissez-vous les bases de Python?
De l'état initial de CentOS8 à l'exécution de php python perl ruby avec nginx
Extraire des images et des tableaux de pdf avec python pour réduire la charge de reporting
Un mémo sur la façon de surmonter le problème difficile de la capture d'effets avec l'IA
Le 15e temps réel hors ligne, j'ai essayé de résoudre le problème de l'écriture avec python
Comment gérer le phénomène que Python (notebook Jupyter) exécuté sur WSL devient abandonné
J'ai essayé de trouver l'entropie de l'image avec python
J'ai essayé de simuler la propagation de l'infection avec Python