GPU-Lehrer Li Mu

import torch
from torch import nn
print(torch.device('cpu'))

print(torch.cuda.device_count())

Ausgabe:

CPU

über

Guess you like

Origin blog.csdn.net/qq_45828494/article/details/126652290