原因是GPU内存不够了.
解决方法有以下几种:
- 减小batch_size
- 只使用CPU不用GPU
import os
os.environ['CUDA_VISIBLE_DEVICES'] = '/gpu:0'
- 修改cuda或者cuDnn的版本. 但是好麻烦. 详细版本见这里.
Reference
[1] Blas xGEMMBatched launch failed的出现原因
原因是GPU内存不够了.
解决方法有以下几种:
import os
os.environ['CUDA_VISIBLE_DEVICES'] = '/gpu:0'
Reference
[1] Blas xGEMMBatched launch failed的出现原因