利用GPU训练时的常见错误

1.CUDA_VIDIBLE_DEVICES=4,5 python3 main.py   我想在集群条件下利用4,5号GPU，因为模型较小，并不清楚是均可以指定4,5号GPU（内存占用状况看不出来。。。后续会找找其余验证办法），报了以下几个错误：python all tensors must be on devices[0]：缓存 CUDNN_STATUS_INTERNAL_ERROR：目前找到了

>>阅读原文<<