pytorch使用cuda报错RuntimeError: CUDA error: unknown error，如何解决？

Question

具体情况如下

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上运行就没有报错。。。
不知道为什么。。。

Answer 1

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