os.environ[“CUDA_VISIBLE_DEVICES“]设置GPU不成功

sol:
需要在import torch之前先设置
os.environ[“CUDA_VISIBLE_DEVICES”] = “6,7”

在import之前需要将数据配置好,否则会使用默认GPU 0

你可能感兴趣的:(问题记录,python,pytorch)