device Pytorch to用法( 二 )


print("Using GPU")
else:
device = torch.device("cpu")
print("Using CPU")
```
2. 确认设备类型和编号
在使用多个GPU时,需要指定设备类型和编号 。例如,如果有2个GPU,可以使用如下代码指定使用第二个GPU:
```python
device = torch.device("cuda:1")
```
3. 确认数据类型是否匹配
在进行数据转移时,需要确保数据类型一致 。例如,如果将float类型的Tensor转移到GPU上,则需要确保GPU支持float类型 。如果不匹配,则会出现类型错误 。
【device Pytorch to用法】四、

推荐阅读