net is a neural network I desired,I want to put it to gpu
In the pytorch tutorial,the code is:
net.to(device)
but when I run this code,an error occurs:
CUDA error: unknown error

image.png636×835 15.4 KB

by the way,my cuda is available
so if anyone could help me with this problem,very thx to you