不使用 GPU 的 HuggingFace 加速模型

问题描述 投票:0回答:0

我正在尝试使用 HuggingFace 模型之一 accelerate 并已按照设置教程步骤进行操作。

我似乎遇到的问题是我使用了加速配置并将我的机器设置为使用我的 GPU,但是在查看资源监视器后我的 GPU 使用率仅为 7% 我不认为我的训练正在使用我的 GPU所有,我有一个 3090TI.

这是我用来开始训练的命令

accelerate launch train_unconditional.py
–train_data_dir=“data”
–resolution=256
–center_crop
–random_flip
–output_dir=“output-256”
–train_batch_size=2
–save_model_epochs=5
–num_epochs=100
–gradient_accumulation_steps=1
–use_ema
–learning_rate=1e-4
–lr_warmup_steps=500
–mixed_precision=no

是否有我缺少的库或我需要更改的其他一些配置设置,因为在加速文档中我只能找到有关多 gpu 训练的信息。

python torchvision huggingface
© www.soinside.com 2019 - 2024. All rights reserved.