Mr_Xuf的博客

在PyCharm中运行模型遇到此个问题：RuntimeError: module must have its parameters and buffers on device cuda:0 (device_ids[0]) but found one of them on device: cpu。翻译过来的意思是：模型的参数和buffer（缓存？）需要在cuda:0上运行，但是只找到设备：CPU。这类问题主要涉及到的是模型运行设备的设置，在多GPU上训练模型时，可能后面报错的错误会是but found o

首先声明我使用的环境Ubuntu18.04.2 + Anaconda3在终端中输入命令anaconda-navigator若是出现如下弹框，，不急不急打开文件（注意：红色字体为用户anaconda安装路径哦！！！）vim /root/anaconda3/lib/python3.7/site-packages/anaconda_navigator/app/s...

整体的思路如下：1、保存每个epoch模型的参数；2、如果在训练时意外终止，，则自定义再次加载意外终止处保存的模型；3、然后一切再开始，，比如接着训练测试什么的，，但是问题来了，，出现错误：RuntimeError: expected device cpu but got device cuda:0。。期望CPU设备而得到的是CUDA（GPU）。。原因：optimizer加载参数时,tensor默认在CPU上，故需将所有的tensor都放到GPU上，否则:在optimizer.s