Pytorch 2.0.1+cu118 无法与 Cuda 12.1 一起使用

Question

我的电脑上安装了 CUDA 12.1，并安装了 Pytorch 2.0.1+cu118。

我正在尝试安装CLIP-FIELDS。当我安装gridencoder“python setup.py install”时，出现以下错误：

The detected CUDA version (12.1) mismatches the version that was used to compile
PyTorch (11.8). Please make sure to use the same CUDA versions.

我在官方pytorch论坛上看到https://discuss.pytorch.org/t/install-pytorch-with-cuda-12-1/174294cuda 12.1与Torch 2.0.1+cu118兼容。

有谁知道崩溃的原因吗？

Answer 1

匹配 CUDA 和 Pytorch 的版本通常是一个难题。
我看到 2 个解决方案：

• 原生安装 CUDA 11.8。这可能会很痛苦，并且会破坏其他 python 安装，在最坏的情况下还会破坏计算机中的图形可视化
• 使用正确版本的 pytorch 和 CUDA 创建 Docker 容器。第一次学习这可能会很痛苦，但这是一项可以用于以后安装 git 存储库的技能。并且不存在危害计算机主 CUDA 的风险。

Answer 2

我在安装flash_attn时遇到了这个问题。我在发布包列表中找到了与我的torch版本、python版本、cuda版本一致的发布包。最后我用pip安装成功了

pip install https://github.com/Dao-AILab/flash-attention/releases/download/v2.3.6/flash_attn-2.3.6+cu118torch2.1cxx11abiTRUE-cp310-cp310-linux_x86_64.whl

我使用cuda 11.8，torch 2.1.1，python 3.10