paddle 使用GPU

paddle paddle 中使用gpu训练

记录一下

使用paddle的device的get和set来获取和设置

import paddle
print(paddle.device.get_device())

如果有gpu的话运行结果如下:
获取设备
之后在代码开始位置按照设备名通过set_device()方法设置设备即可:

paddle.device.set_device('gpu:0') # 把get—device的结果直接复制进去
# 之后就可以正常运行其他内容,如
model = MNIST()
train(model)
  • 8
    点赞
  • 22
    收藏
    觉得还不错? 一键收藏
  • 7
    评论
基于paddle框架的图像超分和降噪实现源码+项目使用说明.zip 基于paddle框架的图像超分和降噪实现源码+项目使用说明.zip 基于paddle框架的图像超分和降噪实现源码+项目使用说明.zip 【资源说明】 该项目是个人毕设项目源码,评审分达到95分,调试运行正常,确保可以运行!放心下载使用。 该项目资源主要针对计算机、自动化等相关专业的学生或从业者下载使用,也可作为期末课程设计、课程大作业、毕业设计等。 具有较高的学习借鉴价值!基础能力强的可以在此基础上修改调整,以实现类似其他功能。 项目建议安装anaconda,在anaconda中配置环境,然后导入到pycharm中运行 ## 1. 模型 * [x] [Unet]() ## 2. 数据准备 ## 3. 训练 此模型支持单机单卡和单机多卡训练,以下使用`UNet`跑`denoise`举例。 ### 3.1 单机单卡 ``` python tools/train.py -c ./configs/denoise/unet_watermark.yaml ``` ### 3.2 单机多卡 ``` python -m paddle.distributed.launch --gpus=0,1,2,3 tools/train.py -c ./configs/denoise/unet_watermark.yaml ``` 所有的训练日志都默认保存在 `./output/UNet/train.log` ## 4. 实验结果 ## 5. 评估 此模型支持单机单卡和单机多卡评估,以下使用`UNet`跑`denoise`举例,生成的模型位置在`/output/UNet/best_model.pdparams` ### 5.1 单机单卡 ``` python tools/eval.py -c ./configs/denoise/unet_watermark.yaml -o Global.pretrained_model=./output/UNet/best_model.pdparams ``` ### 5.2 单机多卡 ``` python -m paddle.distributed.launch --gpus=0,1,2,3 tools/eval.py -c ./configs/denoise/unet_watermark_4gpu.yaml -o Global.pretrained_model=./output/UNet/best_model.pdparams ``` 所有的评估日志都默认保存在 `./output/UNet/eval.log` ## 6. 推理 ### 6.1 模型导出 首先需要导出推理模型,例如训练好的模型参数在`./output/UNet/best_model.pdparams`,命令为 ``` python tools/export_model.py -c ./configs/denoise/unet_watermark.yaml -o Global.pretrained_model=./output/UNet/best_model.pdparams ``` 模型将自动导出到`Global.output_dir/Arch.name/inference`,其中`Global.output_dir`和`Arch.name`均在配置文件`.yaml`中。 ### 6.2 模型推理 模型导出后将使用测试数据集对模型进行推理,例如所有的测试文件都在`./test_data`中,运行命令 ``` python tools/inference.py -c ./configs/denoise/unet_watermark.yaml -o Data.Test.path=./test_data ``` 模型会将推理的结果放入`Global.output_dir/Arch.name/Img`中。

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 7
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值