Sistema Windows 10 instala problema de módulo de tocha Python

1. Descrição:

Encontrei um projeto no GitHub hoje e queria executá-lo. Como resultado, precisei instalar o módulo tocha. Em seguida, pip install torchreporte diretamente um erro e a instalação não terá êxito.
endereço do projeto:

2. Solução:

site oficial do pytorch
Módulo tocha Python

Método 1: consulte o comando de instalação no site oficial:

Comando de instalação:

pip install torch==1.6.0+cu101 torchvision==0.7.0+cu101 -f https://download.pytorch.org/whl/torch_stable.html

Fonte específica:
Instale o círculo verde abaixo. Depois de clicar nele, aparece o comando de instalação. Ele pode ser instalado sob cmd, mas é um pouco lento.

Poço de mineração: finalmente descobri que meu computador com janela não suporta GPU cuda, então preciso usar a versão da cpu:

Comando de instalação:

pip install torch==1.6.0+cpu torchvision==0.7.0+cpu -f https://download.pytorch.org/whl/torch_stable.html

Método 2: Baixe o arquivo para instalar:

O arquivo whl que baixei aqui pode ser instalado diretamente:
download endereço do arquivo:
vá aqui para fazer o download do correspondente e execute

pip install 下载的绝对理解地址（可以直接拖过来）

Três, alguma outra resolução de erro:

Erro 1:untimeError: Attempting to deserialize object on a CUDA device but torch.cuda.is_available() is False. If you are running on a CPU-only machine, please use torch.load with map_location=torch.device('cpu') to map your storages to the CPU.

Referência da solução:

Eu vi esta frase:

my_model = net.load_state_dict(torch.load('classifier.pt', map_location=torch.device('cpu')))

Adicione este parâmetro map_location=torch.device('cpu')à base original