torch与torchvision对应版本
时间: 2023-09-30 13:05:08 浏览: 294
jetson nx等开发板上torch和torchvision的完美解决方案
torch与torchvision是PyTorch深度学习框架的两个常用库。
对应版本的选择需要注意它们与PyTorch版本的兼容性。一般情况下,你需要确保torch和torchvision的版本与你安装的PyTorch版本相匹配。
以下是一些常见的对应版本关系:
- PyTorch 1.7及以上版本建议使用torchvision 0.8.2及以上版本。
- PyTorch 1.6版本建议使用torchvision 0.7.0版本。
- PyTorch 1.5版本建议使用torchvision 0.6.1版本。
- PyTorch 1.4版本建议使用torchvision 0.5.0版本。
请注意,这只是一些常见的对应关系,具体的版本选择还需要结合你的具体使用场景和需求来确定。你可以根据官方文档或者PyTorch社区的建议来选择适合你的版本。
阅读全文