【无标题】pytorch指定GPU无效问题

import os
os.environ[“CUDA_VISIBLE_DEVICES”] = “1”
在指定程序执行的GPU时,上述语句要在import torch之前才可生效。

你可能感兴趣的:(pytorch)