[Python] Pythonコードのトーチバージョンを判断します(以前のバージョンとのコード互換性のために使用されます)

0.環境

ubuntu
torch1.1.0

torch1.7.0+cu110

1。目的

サーバーのトーチバージョンが異なるため、場所によってはエラーが発生しません。

そこで、コードに判断のバージョンを追加することを考えました。

2.ターミナルで実験する

python

import torch
print(torch.__version__)

2つのデバイスが取得します:

1.1.0

1.7.0+cu110

3.コード例

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

 

 

 

 

おすすめ

転載: blog.csdn.net/qq_35975447/article/details/115235560
おすすめ