模型火车系统:AWS Ubuntu p2.xlarge,R 3.4.0,mxnet_1.0.1。 保存通过:
mx.model.save(A3.MXmodel, "Action/A3.MXmodel", iteration = 3000)
在同一系统上加载可以正常工作:
A3.MXmodel <- mx.model.load("A3.MXmodel", iteration=3000)
A3.pred <- predict(A3.MXmodel, as.matrix(nNewVector))
A3.pred.label = max.col(t(A3.pred))-1
将模型文件移动到新系统(g2.xlarge上的第一个实例BUT的AMI克隆)。并试图预测:
A3.pred <- predict(A3.MXmodel, as.matrix(nNewVector))
导致rstudio立即崩溃,没有数据保存或错误消息。 我可以通过安装检查确认mxnet正在处理新实例:
library(mxnet)
a <- mx.nd.ones(c(2,3), ctx = mx.gpu())
b <- a * 2 + 1
b
我是否必须在新的意义上指明模型基于GPU设备? 可以在具有CPU mxnet构建的CPU实例上运行在GPU实例上训练的模型吗?