2020年9月15日
pytorch 在cpu的变量转到GPU的方法
摘要： tensor数据的cuda方法返回变量值的device为cuda，并不会直接移动当前变量到GPU。 举例： B = A.cuda() 其中A为CPU变量，那么执行上面语句后，A依旧在CPU上，创建的新的数据B是A在GPU上面的拷贝，当然单独执行A.cuda()，A也依旧在CPU上面。 import 阅读全文
posted @ 2020-09-15 14:03 半夏生 阅读(1981) 评论(0) 推荐(0) 编辑