pytorch
记录自己学习pytorch遇到的问题,以及总结。
学不会编程的小男孩
每天都要有所作为!
展开
-
pytorch GPU版安装教程(不用安装cuda和cudnn)
写这篇博文的原因是:因为网上有太多教程说安装pytorch GPU版要配套安装相同版本的不用单独安装cuda和cudnn驱动安装包,但如果用anaconda的话,实际上是不用的,anaconda会自动帮我们安装好,全程不需要额外下载其他安装包。安装过程也很方便。下面首先介绍一下我的电脑环境情况:windows 10 系统 anaconda首先,鼠标右键——>NVDIA控制面板——>帮助——>系统信息——>组件,查看自己电脑显卡的CUDA版本然后,访问https:/原创 2020-07-27 12:10:12 · 16215 阅读 · 8 评论 -
Expected object of backend CPU but got backend CUDA for argument #2 ‘weight‘——使用pytorch遇到的异常总结(持续更新)
本博客记录我在学习pytorch的过程中,遇到的一些问题以及解决方法。希望对大家有用。在编写第一个程序时,想尝试把模型代码搬到GPU上训练时出现的错误。Expected object of backend CPU but got backend CUDA for argument #2 'weight'这个错误是:程序需要一个CUDA tensor,我给的却是个CPU tensor。这种问题出现的原因有很多种。比如在程序中创建一个新的tensor,但没指定CUDA,默认是CPU te..原创 2020-07-27 11:17:46 · 1385 阅读 · 0 评论