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