我无法找到 cuda 驱动程序 12.2 的 Pytorch 版本。谁能指导我在哪里可以找到有帮助的材料。
我当前安装的 pytorch 版本为 11.7。在训练模型时,我面临以下错误。
** 运行时错误(CUDA_MISMATCH_MESSAGE.format(cuda_str_version,torch.version.cuda)
检测到的 CUDA 版本 (12.2) 与用于编译 PyTorch (11.7) 的版本不匹配。请确保使用相同的 CUDA 版本。**
PS:我有nvidia驱动程序535
提前致谢