如何在 Pytorch 中检查张量是不是在 cuda 上？
Posted
技术标签:
【中文标题】如何在 Pytorch 中检查张量是不是在 cuda 上？【英文标题】：How to check if a tensor is on cuda in Pytorch?如何在 Pytorch 中检查张量是否在 cuda 上？ 【发布时间】：2021-03-30 12:52:32 【问题描述】：我有一个张量
t = torch.zeros((4, 5, 6))
如何检查它是否在gpu上？
【问题讨论】：
【参考方案1】：From the pytorch forum
使用t.is_cuda
t = torch.randn(2,2)
t.is_cuda # returns False
t = torch.randn(2,2).cuda()
t.is_cuda # returns True
【讨论】：
以上是关于如何在 Pytorch 中检查张量是不是在 cuda 上？的主要内容，如果未能解决你的问题，请参考以下文章
当目标不是单热时，如何计算 Pytorch 中 2 个张量之间的正确交叉熵？
如何在 pytorch 和 tensorflow 中使用张量核心？