import torch #返回当前设备索引# torch.cuda.current_device() #返回GPU的数量# torch.cuda.device_count() #返回gpu名字,设备索引默认从0开始# ...
確定! 回上一頁