(Ubuntu / Windows) verificación de pytorch (si pytorch se instaló correctamente)

Verifique que Pytorch esté disponible (CPU / GPU)

El siguiente es el código de verificación para comprobar si pytorch se instaló correctamente. Si ve que no hay ningún error y el resultado es normal, ¡debe felicitarlo por el primer paso de Xiaobai!
Compártelo: después de confirmar tu propio controlador de tarjeta gráfica, puedes instalar las dos versiones de pytorch (GPU / CPU) directamente a través del comando en el sitio web oficial de Pytorch. ¡La falla de instalación es básicamente un problema de red!
pip install -i "Douban Source" ---- >>>> (se puede hacer de esta manera general u)

Prueba en GPU

import torch   # 能否调用pytorch库

print(torch.cuda.current_device())   # 输出当前设备（我只有一个GPU为0）
print(torch.cuda.device(0))   # <torch.cuda.device object at 0x7fdfb60aa588>
print(torch.cuda.device_count())  # 输出含有的GPU数目
print(torch.cuda.get_device_name(0))  # 输出GPU名称 --比如1080Ti
x = torch.rand(5, 3)
print(x)  # 输出一个5 x 3 的tenor(张量)

Prueba en CPU

import torch
x = torch.rand(5,3)
print(x)