报错cudnn的问题:
return F.conv2d(input, weight, bias, self.stride,
RuntimeError: cuDNN error: CUDNN_STATUS_INTERNAL_ERROR
You can try to repro this exception using the following code snippet. If that doesn't trigger the error, please include your original repro script when reporting this issue.
总是再net.forward()这里报错,可能是环境的问题,但是这个环境用的是好好的,cudnn版本与torch都对应,就是报错,尝试了好几种办法,最后简单的一句命令解决。
解决:
CUDA_VISIBLE_DEVICES=1 python inference.py
加一行GPU号指定就可以了,完美!!!