0. Meio Ambiente
ubuntu
torch1.1.0
torch1.7.0+cu110
1. Propósito
Como a versão da tocha do servidor é diferente, não haverá erros em alguns lugares.
Então, pensei em adicionar uma versão de julgamento ao código.
2. Experimente no terminal
python
import torch
print(torch.__version__)
Dois dispositivos obterão:
1.1.0
1.7.0+cu110
3. Exemplo de código
import torch
if torch.__version__ <= "1.1.0"
pass
if torch.__version__ == "1.7.0+cu110"
pass