问答

关于参数在CPU和GPU的问题

作者:admin 2021-04-18 我要评论

作为一个新手,我问的东西可能有点基础,但是确实没有在网上找到解决方案,请各位前辈包涵。 https://tangshusen.me/Dive-in... https://tangshusen.me/Dive-in....

在说正事之前,我要推荐一个福利:你还在原价购买阿里云、腾讯云、华为云服务器吗?那太亏啦!来这里,新购、升级、续费都打折,能够为您省60%的钱呢!2核4G企业级云服务器低至69元/年,点击进去看看吧>>>)

作为一个新手,我问的东西可能有点基础,但是确实没有在网上找到解决方案,请各位前辈包涵。


https://tangshusen.me/Dive-in...


https://tangshusen.me/Dive-in...

我在上面这个网址学习pytorch,跟着运行demo的时候出现了如下的报错预计。一行代码都没有改,却出现了这样的问题。
image.png

RuntimeError: Tensor for 'out' is on CPU, Tensor for argument #1 'self' is on CPU, but expected them to be on GPU (while checking arguments for addmm)

网上有些解决方法,但是目前知识不够,确实没法看懂。

###

tensor可以相互运算的前提是,它们处在同一个设备中,所以如果你无法确定tensor的位置的话,可以在运算之前统一移动到cpu上(tensor = tensor.cpu())或统一移动到gpu上(tensor = tensor.to(torch.device("cuda:0"))),再进行运算。

版权声明:本文转载自网络,遵循 CC 4.0 BY-SA 版权协议,转载请附上原文出处链接和本声明。本站转载出于传播更多优秀技术知识之目的,如有侵权请联系QQ/微信:153890879删除

相关文章
  • nginx响应速度很慢

    nginx响应速度很慢

  • 点击选中的多选框,会在已选那一栏显示

    点击选中的多选框,会在已选那一栏显示

  • PHP 多态的理解

    PHP 多态的理解

  • 关于C语言中static的问题

    关于C语言中static的问题

腾讯云代理商
海外云服务器