import torch #交互模式运行代码出现以下错误
OSError: /usr/lib/aarch64-linux-gnu/libcudnn.so.8: file too short
解决方案:
apt-get install libcudnn8
#交互模式运行以下代码
import torch
torch.cuda.is_available()
>>True
torch.zeros((1,2,3,4), device=0) #运行此行代码出现以下错误
RuntimeError: CUDA error: no kernel image is available for execution on the device
CUDA kernel errors might be asynchronously reported at some other API call,so the stacktrace belo