Instale o pytorch no servidor remoto e instale-o de acordo com o comando do site oficial, mas após a conclusão mostra que a GPU está indisponível, então registre este pit.
1. Instale de acordo com o site oficial
A instalação é rápida, no entanto! ! Após o término da instalação, digite o seguinte código para verificar a instalação, mas mostra que a instalação não foi bem-sucedida! !
import torch # 如果pytorch安装成功即可导入
print(torch.cuda.is_available()) # 查看CUDA是否可用
print(torch.cuda.device_count()) # 查看可用的CUDA数量
print(torch.version.cuda) # 查看CUDA的版本号
Portanto, isso indica uma grande falha na instalação!
Mas não desisti e digitei o seguinte depoimento para verificar a tocha
conda list
Isso mostra que estamos instalando apenas a versão CPU! !
Dois, a solução
Por causa desse grande buraco, pode-se dizer que uma noite inteira foi perdida. Depois de aprender com muitos blogs, o seguinte blog do blogueiro pode ser considerado uma solução super boa.
Seguindo as orientações deste blog, a versão GPU do torch foi instalada com sucesso usando pip
por meio deste registro