pytorch使用cuda报错RuntimeError: CUDA error: unknown error,如何解决?

具体情况如下

cuda可用但是把tensor放到gpu上就会报这个错

import torch as t
tensor=t.Tensor(3,4)
tensor.cuda(0)

按照类似问题https://ask.csdn.net/questions/767989所给的方法,检查了cuda版本,全局设置全局设置首选图形处理器为“高性能NVIDIA处理器”
但还是报错

显卡是gtx1050
按照网上方法运行cuda自带的deviceQuery.exe的结果

图片说明

求助!

更新7.14 13:56

代码

import torch as t
tensor=t.Tensor(3,4)
tensor.cuda(0)

在pycharm上运行就没有报错。。。
不知道为什么。。。

3个回答

我是数据集小的时候可以训练,但是数据变大(和我之前的没有GPU时数据量一致)就出现这个错误了,重装驱动和cuda+cudnn还是不能解决

检查下你用的pytorch是不是gpu版本的,并且和你的cuda/cudnn的版本是否匹配。

davidlucio
davidlucio 额我后来重装了,问题就解决了
大约一年之前 回复

我把显卡驱动,cuda,cudnn,pyTorch全部重装一遍,然后就没报错了。。。

Csdn user default icon
上传中...
上传图片
插入图片
抄袭、复制答案,以达到刷声望分或其他目的的行为,在CSDN问答是严格禁止的,一经发现立刻封号。是时候展现真正的技术了!
立即提问