pytorch 总是遇到cuda out of memery,4块卡,其中3块一样,另外一块不一样。。。

一般用法:(4块卡完全一样)

CUDA_VISIBLE_DEVIS=0 python train.py

对于不一样算力的显卡来说,优先使用算力高的,

除非使用CUDA_DEVICE_ORDER=PCI_BUS_ID,让其按照物理排序,而不是按照算力优先级排序

CUDA_DEVICE_ORDER=PCI_BUS_ID CUDA_VISIBLE_DEVIS=0 python train.py就行了

你可能感兴趣的:(杂货,pytorch,pytorch)