pytorch出现:CUDA error: CUBLAS_STATUS_INTERNAL_ERROR when calling `cublasSgemm( handle, opa, opb, m, n

主要原因是多GPU,但是我们使用单卡训练的问题(我是这样的问题)

CUDA_VISIBLE_DEVICES="你的gpu号"  python main.py --device="cuda:0"

你可能感兴趣的:(pytorch,python,人工智能)