pytorch的代码 CPU改为GPU

pytorch的代码 CPU改为GPU
找一下几个地方改动

网络模型
数据（输入、标记）
损失函数
cuda()

一、网络模型

mymodule = MyModule()
mymodule = mymodule.cuda()

二、数据（输入、标记）

        images,targets = dataimages = images.cuda()targets = targets.cuda() #训练数据和测试数据集都要进行此操作

三、损失函数

# 损失函数
loss = nn.CrossEntropyLoss()
loss = loss.cuda()

上面的写法，在无GPU时会报错。更好的写法是有GPU用GPU，无GPU用CPU

mymodule = MyModule()
if torch.cuda.is_available():mymodule = mymodule.cuda()

pytorch的代码 CPU改为GPU相关推荐

Tensorflow + PyTorch 安装（CPU + GPU 版本）
目录一.Anaconda 安装二.安装 TensorFlow-CPU 1.配置环境 2.安装 Tensorflow 三.安装TensorFlow-GPU 1.是否可安装GPU版Tensorflow ...
Pytorch：比较CPU和GPU的运算速度
在跑神经网络的时候,GPU的作用是很明显的.下面比较一下CPU跑和GPU跑的区别: 先用CPU跑: import torch import timefor i in range(1,10):start ...
PyTorch 笔记（03）— Tensor 数据类型分类（默认数据类型、CPU tensor、GPU tensor、CPU 和 GPU 之间的转换、数据类型之间转换）
1. Tensor 数据类型 Tensor 有不同的数据类型,如下表所示,每种类型都有 CPU 和 GPU 版本(HalfTensor)除外,默认的 tensor 是数据类型是 FloatTensor ...
收藏！PyTorch常用代码段合集
↑↑↑关注后"星标"Datawhale 每日干货 & 每月组队学习,不错过 Datawhale干货作者:Jack Stark,来源:极市平台来源丨https://zhu ...
PyTorch常用代码段合集
↑ 点击蓝字关注视学算法作者丨Jack Stark@知乎来源丨https://zhuanlan.zhihu.com/p/104019160 极市导读本文是PyTorch常用代码段合集,涵盖基本 ...
【深度学习】PyTorch常用代码段合集
来源 | 极市平台,机器学习算法与自然语言处理本文是PyTorch常用代码段合集,涵盖基本配置.张量处理.模型定义与操作.数据处理.模型训练与测试等5个方面,还给出了多个值得注意的Tips,内容非常 ...
pytorch list转tensor_PyTorch 52.PyTorch常用代码段合集
本文参考于: Jack Stark:[深度学习框架]PyTorch常用代码段zhuanlan.zhihu.com 1. 基本配置导入包和版本查询: import torch import torc ...
收藏 | PyTorch常用代码段合集
点上方蓝字计算机视觉联盟获取更多干货在右上方 ··· 设为星标 ★,与你不见不散仅作学术分享,不代表本公众号立场,侵权联系删除转载于:作者丨Jack Stark@知乎来源丨https://zh ...
深度盘点：PyTorch常用代码段合集
本文是PyTorch常用代码段合集,涵盖基本配置.张量处理.模型定义与操作.数据处理.模型训练与测试等5个方面,还给出了多个值得注意的Tips,内容非常全面. PyTorch最好的资料是官方文档.本文 ...

pytorch的代码 CPU改为GPU

pytorch的代码 CPU改为GPU相关推荐

最新文章

热门文章