PyTorch 10.2和CUDA 11.2兼容吗
时间: 2024-05-21 07:11:27 浏览: 117
Pytorch1.11_CUDA11.3_Pycharm2022_调试环境搭建
不完全兼容。PyTorch 1.0.2要求CUDA版本为10.0或10.1,而CUDA 11.2是一个较新的版本。虽然在某些情况下可能能够在PyTorch 1.0.2中使用CUDA 11.2,但这并不是官方支持的组合,因此使用时需要小心。如果需要使用CUDA 11.2,建议使用PyTorch的最新版本(目前是1.9.0)。
阅读全文