Torch not compiled with CUDA enabled

RubyLinT

已于 2024-01-12 08:48:32 修改

阅读量659

点赞数 10

文章标签： chatgpt python

于 2024-01-11 17:06:48 首次发布

本文链接：https://blog.csdn.net/RubyLinT/article/details/135533963

版权

作者在尝试使用chatglm3对话模型时遇到CUDA兼容性问题，因为其Intel显卡不支持CUDA。文章描述了作者的探索过程，包括检查GPU可用性、驱动安装失败以及找到的解决方案（如使用云服务器）。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

最近接触chatglm3对话预训练模型，从git上下载，装包装半天，最后终于跑起来了，但是一对他进行对话，后台就开始报错了

File "E:\Python311\Lib\site-packages\torch\nn\modules\linear.py", line 114, in forward return F.linear(input, self.weight, self.bias)

反正大概是torch这个包有问题，于是就单独写个torch的代码执行一下，代码如下：

import torch
print("是否可用：", torch.cuda.is_available())        # 查看GPU是否可用
print("GPU数量：", torch.cuda.device_count())        # 查看GPU数量
print("torch方法查看CUDA版本：", torch.version.cuda)  # torch方法查看CUDA版本
print("GPU名称：", torch.cuda.get_device_name())    # 根据索引号得到GPU名称
print("GPU索引号：", torch.cuda.current_device())    # 查看GPU索引号

然后就报错：说我的GPU显卡不可用