Pytorch学习--神经网络--利用GPU训练

全部代码依托于该博客

一、第一种方法:XX.cuda()

模型、数据+标签、损失函数都放到 gpu 上面

代码更改如下:

二、第二种方法:XX.to(device) --更推荐

device = torch.device("cuda" if torch.cuda.is_available() else "cpu")

同样将模型、数据+标签、损失函数都放到 device 上面

代码更改如下:



相关推荐
yusaisai大鱼3 分钟前
tensorflow_probability与tensorflow版本依赖关系
人工智能·python·tensorflow
18号房客3 分钟前
一个简单的深度学习模型例程,使用Keras(基于TensorFlow)构建一个卷积神经网络(CNN)来分类MNIST手写数字数据集。
人工智能·深度学习·机器学习·生成对抗网络·语言模型·自然语言处理·tensorflow
Biomamba生信基地7 分钟前
R语言基础| 功效分析
开发语言·python·r语言·医药
神秘的土鸡11 分钟前
神经网络图像隐写术:用AI隐藏信息的艺术
人工智能·深度学习·神经网络
虾球xz12 分钟前
游戏引擎学习第58天
学习·游戏引擎
数据分析能量站12 分钟前
神经网络-LeNet
人工智能·深度学习·神经网络·机器学习
Jaly_W20 分钟前
用于航空发动机故障诊断的深度分层排序网络
人工智能·深度学习·故障诊断·航空发动机
CodeClimb22 分钟前
【华为OD-E卷-木板 100分(python、java、c++、js、c)】
java·javascript·c++·python·华为od
小嗷犬22 分钟前
【论文笔记】Cross-lingual few-shot sign language recognition
论文阅读·人工智能·多模态·少样本·手语翻译
夜幕龙29 分钟前
iDP3复现代码数据预处理全流程(二)——vis_dataset.py
人工智能·python·机器人