[Python] Juzga la versión de la antorcha en código Python (usado para compatibilidad de código con versiones menores)

0. Medio ambiente

ubuntu
torch1.1.0

torch1.7.0+cu110

1. Propósito

Debido a que la versión de antorcha del servidor es diferente, no habrá errores en algunos lugares.

Entonces pensé en agregar una versión de juicio al código.

2. Experimento en la terminal

python

import torch
print(torch.__version__)

Dos dispositivos obtendrán:

1.1.0

1.7.0+cu110

3. Ejemplo de código

import torch
if torch.__version__ <= "1.1.0"
    pass
if torch.__version__ == "1.7.0+cu110"
    pass