Instale pytorch en el servidor remoto y siga los comandos del sitio web oficial, pero una vez completado, muestra que la GPU no está disponible, por lo que se registra este error.
1. Instalar según el sitio web oficial.
¡Sin embargo, la instalación es rápida! ! Una vez completada la instalación, ingresé el siguiente código para verificar la instalación, ¡pero mostró que la instalación no fue exitosa! !
import torch # 如果pytorch安装成功即可导入
print(torch.cuda.is_available()) # 查看CUDA是否可用
print(torch.cuda.device_count()) # 查看可用的CUDA数量
print(torch.version.cuda) # 查看CUDA的版本号
¡Esto indica un gran error en la instalación!
Pero sin rendirme, ingresé la siguiente declaración para verificar la antorcha.
conda list
¡Esto muestra que estamos instalando solo la versión de CPU! !
2. Solución
Debido a este gran hoyo, se puede decir que se desperdició toda una noche. Después de aprender de muchos blogs, se puede decir que el siguiente blog del blogger es una solución súper buena.
Siga las instrucciones de este blog y use pip para instalar con éxito la versión GPU de torch.
Por la presente queda registrado