小编ana*_*s r的帖子

cuda 12.2 的 Pytorch 版本

我无法找到 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

提前致谢

gpu version nvidia pytorch

11
推荐指数
1
解决办法
3万
查看次数

标签 统计

gpu ×1

nvidia ×1

pytorch ×1

version ×1