torch使用gpu

在PyTorch中，可以使用以下方法来在GPU上使用torch：
1. 通过使用`torch.cuda.device('cuda')`来指定要使用的GPU。即使没有显式指定要使用哪个GPU，PyTorch默认会将tensor分配在GPU 0上。
2. 默认情况下，PyTorch会在CPU上进行模型训练。可以使用`torch.device('cpu')`来指定使用CPU。
3. 可以通过使用`torch.cuda.current_device()`来查看当前正在使用的GPU的序号。
4. 如果想要尝试使用GPU，但不确定是否存在可用的GPU，可以使用以下两个函数：
- `try_gpu(i=0)`函数可以返回指定序号的GPU（如果存在），否则返回CPU。
- `try_all_gpus()`函数可以返回所有可用的GPU（如果存在），否则返回CPU。