[Python] Julgue a versão da tocha no código Python (usado para compatibilidade de código com versões inferiores)

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

 

 

 

 

Acho que você gosta

Origin blog.csdn.net/qq_35975447/article/details/115235560
Recomendado
Clasificación