Open Kinglee90 opened 1 year ago
return F.conv2d(input, weight, bias, self.stride, RuntimeError: Unable to find a valid cuDNN algorithm to run convolution
应该是超显存了吧,你把bs调小试试
return F.conv2d(input, weight, bias, self.stride, RuntimeError: Unable to find a valid cuDNN algorithm to run convolution![image](https://user-images.githubusercontent.com/66666512/227873006-687c30d1-1831-4ec3-abe7-9476190bb106.png)