I am using pytorch 1.5.0 and my cuda version is 10.2 on ubuntu 18.0. Screenshot from 2020-10-04 01-37-15786×527 72.3 KB
I used resrnet34:
def resnet34(num_classes, grayscale):
“”“Constructs a ResNet-34 model.”""
model = ResNet(block=BasicBlock,
layers=[3, 4, 6, 3],
num_classes=num_classes,
grayscale=grayscale)
return model
Now when I run my script on a small dataset it work perfect. but the same script on the large dataset caused a different error in the line: impf=impf.to(DEVICE) …RuntimeError: CUDA error: unspecified launch failure .