Résoudre le problème de mémoire insuffisante lors de l'exécution de Pytorch
Version:
python: 3.7
pytorch: 1.2
cuda: 10.1
Mémoire vidéo insuffisante lors de l'apprentissage de la migration du pytorch
C’est
Utilisez nvidia-smi pour afficher les informations du processeur (vous devez d'abord ajouter C: \ Program Files \ NVIDIA Corporation \ NVSMI au chemin)
J'ai découvert qu'en fait ma mémoire vidéo est suffisante
On dit que le tensorflow peut entrer en conflit avec le pytorch
Essayez de désinstaller tensorflow-gpu
Invalide
Entrez le code dans le programme
import os
os.environ['CUDA_VISIBLE_DEVICES]='2,3'
Le programme s'exécute avec succès