完成所有安装步骤后，CUDA 在 PyTorch 中不可用

Posted 2023-03-29

【中文标题】完成所有安装步骤后，CUDA 在 PyTorch 中不可用

【英文标题】：CUDA not available in PyTorch after following all installation steps

【发布时间】：2022-01-04 01:03:36

【问题描述】：

我正在尝试让我的 BERT 变压器模型在 CUDA 上运行，并已按照此处的所有安装步骤进行操作：

https://medium.com/@jjlovesstudying/python-cuda-set-up-on-windows-10-for-gpu-support-78126284b085

但是在将文件夹添加到 PATH 变量后，我重新启动 Pycharm 并运行以下命令：

torch.cuda.is_available()

带来False。我很欣赏这是不可重现的，但有人知道如何调试这个问题或解决这个问题吗？需要任何额外信息，我很乐意提供。

【问题讨论】：

这能回答你的问题吗？ Torch.cuda.is_available() keeps switching to False

中差，文档好 - 您是否按照此处的步骤操作？：pytorch.org/get-started/locally（单击字段以构建安装行）

@ti7 这不会在 Windows 上安装 CUDA。我的 cudnn 已安装并且很好，但至于 CUDA 在调用 torch.version.cuda 时是 None？我已经运行了几次安装程序，所以我真的迷路了。

【参考方案1】：

您是否检查过每个部分都加载了模块？ import torch 有什么好处

【讨论】：

加载正常。