cuda12.5应该选择pytorch什么版本
时间: 2024-09-04 09:00:50 浏览: 120
win11及cuda121环境下pytorch安装及避坑
CUDA 12.5通常对应PyTorch的一个较老的稳定分支,因为新版本的CUDA往往会支持最新版的PyTorch。对于CUDA 12.5,推荐使用的PyTorch版本可能是1.8.x到1.9.x之间,因为这是它们官方文档和支持期内对CUDA 12系列有良好兼容性的版本。
然而,需要注意的是,随着时间的推移,一些新的功能和优化可能只存在于更新的版本中,因此如果你需要最新的特性和性能,可能需要尝试1.10.x版本,但可能会面临一些兼容性挑战。建议查看PyTorch官网的CUDA版本支持矩阵,以获取最准确的信息。
阅读全文