如何启用CUDA以使用pytorch？

我有一个基于CUDA的GPU，有384个CUDA内核。但如果我命令torch.cuda.U可用，我仍然无法实现。我想知道如何连接pytorch和CUDA

我试着安装CUDA工具包

torch.cuda.U是否可用

期望 torch.cuda.U是否可用 真的

实际的 torch.cuda.U是否可用

---

False

正如CUDA网页上清楚解释的，您必须检查图形卡的兼容性。例如，“CUDA Toolkit 9”需要至少3的计算能力。 您可以在页面上查看视频卡规格

---

Pytork支持toolkit 9中的CUDA，因此您需要一个兼容的设备。

v0.4之后的Pytork新版本不支持一些较旧的Nvidia GPU。所以，试着在v0.4之前使用PyTorch v0.3。