为pytorch设置可见GPU

在程序代码最前面加上

import os

os.environ["CUDA_VISIBLE_DEVICES"] = "0,1"  

# 例子>> "0"：只有0号GPU可见；"1,3"：只有1号和3号GPU可见；""：全部不可见，此时 torch.cuda.is_available() 输出 False。

注意：不一定非要写在最前面，但至少必须要在 import torch 之前，否则可能不起作用。