问题描述:
pytorch使用GPU报错,具体操作的简化如下:
安装成功,测试GPU可用,但是下面这步报错了。
t = torch.ones(1,1,28,28)
t = t.cuda()
t
我的环境是:
- pytorch1.5.1;
- Cuda 10.1.0;
- GPU(Navida GeForce 920M
- python(3.6)
原因:
问题出在CUDA对部分计算操作无法支持,即no kernel image is available for execution on the device的问题。
可能是GPU和CUDA版本不匹配,也就是硬件的问题,其中,PyTorch 1.3版本,提高了对CUDA设备的Compute Capability等级的最低要求**(计算力3.5及3.5以下的都不支持了)**,导致在较为老旧的GPU型号上,部分计算操作无法执行。
如何查看自己GPU的计算力是否符合要求
在matlab中输入gpuDevice
即可查看:我的是下面这个
我的GPU计算力是3.5是不符合要求的,所以我只能解决软件问题----降低pytorch版本(建议1.2)
安装pytorch可参考:https://blog.csdn.net/peacefairy/article/details/108027467