判断张量是否在GPU上

如何将自己的张量或模型放到GPU去计算:

        1 如果本来就确定有GPU可用,可以直接使用 .cuda()命令进行移动

        2 如果想先检查一下自己的设备,可以用

                device = torch.device("cuda" if torch.cuda.is_available() else "cpu") 进行设备判断,接着利用.to(device)后缀进行张量的移动

两个在Gpu上的张量经过简单的操作生成的新的张量还在Gpu上吗?

        是的,还在Gpu上,想一些简单的拼接相加操作不会改变新张量所处的位置。

        

两个Gpu经过一个没有在Gpu上的神经网络模型,张量还在Gpu上吗

        事实上,这是会报错的,因为在张量进入神经网络模型时,需要判断网络与张量是否在同一台设备中,如果既有cpu又有gpu,就会报错。如果都在Gpu上,出来的也还是在Gpu上。

相关推荐

  1. 判断是否GPU

    2024-01-30 05:08:04       59 阅读
  2. Unity判断鼠标是否UI

    2024-01-30 05:08:04       21 阅读
  3. PyTorch|运算中使用GPU

    2024-01-30 05:08:04       70 阅读
  4. 判断三点是否一直线

    2024-01-30 05:08:04       44 阅读
  5. 【功能】UGUI判断是否点击UI

    2024-01-30 05:08:04       25 阅读

最近更新

  1. docker php8.1+nginx base 镜像 dockerfile 配置

    2024-01-30 05:08:04       94 阅读
  2. Could not load dynamic library ‘cudart64_100.dll‘

    2024-01-30 05:08:04       100 阅读
  3. 在Django里面运行非项目文件

    2024-01-30 05:08:04       82 阅读
  4. Python语言-面向对象

    2024-01-30 05:08:04       91 阅读

热门阅读

  1. VUE中一些概念的理解

    2024-01-30 05:08:04       53 阅读
  2. MyBaties-增删查改

    2024-01-30 05:08:04       59 阅读
  3. Tik Tok帳戶註冊中海外動態IP的作用

    2024-01-30 05:08:04       42 阅读
  4. SpringBoot 实现定时任务

    2024-01-30 05:08:04       59 阅读
  5. mysql学习打卡day19

    2024-01-30 05:08:04       56 阅读
  6. 深入了解 LCC:智能驾驶的关键技术

    2024-01-30 05:08:04       53 阅读
  7. 深入理解vue相关的底层原理

    2024-01-30 05:08:04       46 阅读
  8. Qt程序设计-半透明遮罩效果窗口实现

    2024-01-30 05:08:04       56 阅读