Pytorch CUDA

Question

有关Pytorch与CUDA版本的问题
在Pycharm终端执行以下代码的时候报错

git clone https://github.com/facebookresearch/detectron2 detectron2_repo
pip install -e detectron2_repo

如下图所示：

但是按照我查的资料CUDA12.0就是对应这个版本 请教各位该如何处理

Answer 1

以下内容由CHATGPT及阿里嘎多学长共同生成、有用望采纳：

---

从您提供的截图信息来看，错误提示是说检测到的CUDA版本(11.0)与编译PyTorch时使用的版本(11.8)不匹配。这意味着您需要安装与PyTorch版本相匹配的CUDA版本。

以下是一些解决步骤：

1. 确认PyTorch版本： 首先，确认您安装的PyTorch版本。可以通过运行以下命令来检查：

   python -c "import torch; print(torch.__version__)"
   

2. 安装正确版本的CUDA： 根据PyTorch的官方文档，找到与您的PyTorch版本相匹配的CUDA版本。您可以访问PyTorch的官方网站，查看支持的CUDA版本：

   • PyTorch官方文档

3. 安装CUDA Toolkit： 下载并安装与PyTorch版本相匹配的CUDA Toolkit。您可以从NVIDIA的官方网站下载：

   • NVIDIA CUDA Toolkit

4. 设置环境变量： 安装CUDA Toolkit后，确保将CUDA的bin目录添加到您的PATH环境变量中。

5. 重新安装PyTorch（如果需要）： 如果您更改了CUDA版本，可能需要重新安装PyTorch以确保它与新的CUDA版本兼容。

6. 重新尝试安装Detectron2： 在确保CUDA和PyTorch版本匹配后，重新尝试安装Detectron2。

如果上述步骤无法解决问题，您可能需要查看Detectron2的文档或GitHub仓库的issue部分，看看是否有其他用户遇到类似问题，并找到解决方案。

此外，以下是一些可能有用的参考资料：

• Detectron2 GitHub仓库
• Detectron2官方文档

请注意，由于我无法访问外部链接，上述链接需要您自行访问。希望这些信息对您有所帮助！