请问: 运行本项目的模型对电脑硬件有要求吗？

[michael7908]

我用的是Win10，而且是非常老旧的组装机，没有高端显卡。

谢谢林总的精彩分享。

[linyiLYi]

运行 test.py 跑测试对算力要求并不高，因为老游戏硬件需求很小，Nature 卷积网络的需求也不大，使用普通处理器应该就可以了。如果想使用 CPU 运行，可以在 test.py 文件里搜索“cuda”将其删去，“cuda”会要求程序使用 GPU 来运行，删去后就会选择 CPU 来运行了

[leesonchen]

我跑train.py时候提示Using cpu device，而且GPU的使用率确实也是0%，请问如何调整使其用GPU训练呢？

[xuanquanchen]

可能是pytorch版本不对 首先检查gpu是否可用，运行下面的python代码

import torch

device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
print("Using device:", device)

如果显示cpu 在终端中输入

nvidia-smi

在输出的右上角能找到cuda版本 卸载pytorch，去pytorch官网(https://pytorch.org) 查看对应cuda版本的pytorch并安装

[Datou]

我跑train.py时候提示Using cpu device，而且GPU的使用率确实也是0%，请问如何调整使其用GPU训练呢？

应该是安装了非CUDA版的torch，先把这个torch卸载掉，然后安装CUDA版的torch。 pip uninstall torch pip install torch --index-url https://download.pytorch.org/whl/cu118

现在一般是装这个cu118版本，具体的安装命令去pytorch官网查询最保险。

[lslzl3000]

我跑train.py时候提示Using cpu device，而且GPU的使用率确实也是0%，请问如何调整使其用GPU训练呢？

应该是安装了非CUDA版的torch，先把这个torch卸载掉，然后安装CUDA版的torch。 pip uninstall torch pip install torch --index-url https://download.pytorch.org/whl/cu118

现在一般是装这个cu118版本，具体的安装命令去pytorch官网查询最保险。

大概训练的时间是多少呢

[Datou]

选好参数后几个小时就能出结果，但选参数可能要花一两周。