我找不到Cuda驱动程序12.2的Pytorch版本。有谁能告诉我在哪里可以找到任何有用的材料吗？我已经安装了pytorch 11.7版。在训练模型时，我遇到了以下错误。
**运行时错误（CUDA_MISMATCH_MESSAGE.format（CUDA_str_version， Torch .版本.CUDA））
检测到的CUDA版本（12.2）与编译PyTorch所用的版本（11.7）不匹配。请确保使用相同的CUDA版本。**PS：我有nvidia驱动程序535先谢了
htzpubme1#
我也有同样的问题。目前，PyTorch的CUDA 12.2版本尚未发布。以下是我目前正在探索的几种选择。我还没决定用哪一个：1.使用NGC容器https://catalog.ngc.nvidia.com/orgs/nvidia/containers/pytorch这本质上就像一个虚拟环境，包含所需的CUDA驱动程序和所有内容。1.降级到CUDA 12.1并安装夜间版本https://download.pytorch.org/whl/nightly/cu121的不确定确切的驱动程序版本，我会需要这一点，但它可能是一个可行的选择。1.只是等待并希望他们能尽快发布1.（最有可能的是）我只会从源头构建它。https://github.com/pytorch/pytorch/issues/104417的
参考文献
https://discuss.pytorch.org/t/which-version-of-cuda-12-1-should-be-used/176456https://discuss.pytorch.org/t/pytorch-for-cuda-12/169447/2
1条答案
按热度按时间htzpubme1#
我也有同样的问题。目前，PyTorch的CUDA 12.2版本尚未发布。
以下是我目前正在探索的几种选择。我还没决定用哪一个：
1.使用NGC容器
https://catalog.ngc.nvidia.com/orgs/nvidia/containers/pytorch
这本质上就像一个虚拟环境，包含所需的CUDA驱动程序和所有内容。
1.降级到CUDA 12.1并安装夜间版本
https://download.pytorch.org/whl/nightly/cu121的
不确定确切的驱动程序版本，我会需要这一点，但它可能是一个可行的选择。
1.只是等待并希望他们能尽快发布
1.（最有可能的是）我只会从源头构建它。https://github.com/pytorch/pytorch/issues/104417的
参考文献
https://discuss.pytorch.org/t/which-version-of-cuda-12-1-should-be-used/176456https://discuss.pytorch.org/t/pytorch-for-cuda-12/169447/2