0.环境
ubuntu
torch1.1.0
torch1.7.0+cu110
1.目的
因为服务器的torch版本不一样,有些地方会无法出错。
所以想到在代码中加入一个版本的判断。
2.终端中实验
python
import torch
print(torch.__version__)
两台设备分别会得到:
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