当前位置： 首页 > article >正文

pytorch中数据和模型都要部署在cuda上面

article 2024/11/16 7:28:37

注意一定要两个都部署在模型训练之前，缺少一个都会报错在两个设备上训练，一个在cpu，一个在cuda

部署数据到cuda上

point_features = torch.tensor(point_features, dtype=torch.float32).to('cuda')  # 如果你有支持的 GPU,shape = (499，3159,3)
coord_time = torch.tensor(coord_time, dtype=torch.float32).to('cuda')  # 如果你有支持的 GPU  shape = (3159,3)
target = torch.tensor(target, dtype=torch.float32).to('cuda')  # 如果你有支持的 GPU  shape = (499，3159,3)

部署模型到cuda上  

   
model = model.to('cuda')


http://www.kler.cn/a/395525.html

相关文章：

  • git命令提交项目
  • mysql每日一题（上升的温度,date数据的计算）
  • 从H264视频中获取宽、高、帧率、比特率等属性信息
  • CSS的综合应用例子（网页制作）
  • AutoDL远程连接技巧
  • 算法每日双题精讲——滑动窗口（长度最小的子数组，无重复字符的最长子串）
  • ctfshow-web入门-JWT（web345-web350）
  • 电动车租赁支付宝免押小程序开发方案php+uniapp
  • vue项目PC端和移动端实现在线预览pptx文件
  • YOLOv7-0.1部分代码阅读笔记-metrics.py
  • 三正科技笔试题
  • 【linux】centos7 换阿里云源
  • 调用 Xinference OpenAI接口时报错 Model not found in the model list, uid
  • 前端埋点、监控
  • 如何保证MySQL与Redis缓存的数据一致性？
  • PC上浏览器是如何查询DNS 缓存的呢？
  • 自建k8s集群，利用开源的GitLab、Jenkins和Harbor实现CI/CD和DevOps的过程回顾
  • Redis 概 述 和 安 装
  • C++初阶——stack
  • 服务器被挂马怎么办？——解决服务器被挂马的方法和步骤
  • 10款录屏工具个人使用感分享！！！！！！
  • 用MVVM设计模式提升WPF开发体验：分层架构与绑定实例解析
  • 深度学习之GAN应用
  • Springboot 整合 itext 实现PDF文件合并，识别图片则转成PDF拼接
  • C++图案例大全
  • 算法----阶乘问题