本项目如何使用GPU进行训练。谢谢各位大佬！大佬发财！
goliouu opened this issue · comments
(StreetFighterAI) D:\街霸ai\street-fighter-ai\main>python train.py
Using cpu device
Wrapping the env in a VecTransposeImage.
目前的训练是使用cpu来进行的，如何使用GPU来进行？
同问，本地已安装 tensorflow_gpu，测试脚本是能看到 gpu。
train.py，有将 device设置成 cuda，但是还是没有效果。
进入PPO文件可以看到，引入了 torch，所以需要再conda环境里面，安装torch，否则会默认 cpu 训练。
安装torch方法，要注意你本地 cuda 版本，版本对应关系如下：https://pytorch.org/get-started/previous-versions/
最后加一段测试代码，到 ppo.py，可以查看GPU是否启用。
if th.cuda.is_available(): print("CUDA is available!") else: print("CUDA is not available.")
执行结果如下：
先看看PyTorch支不支持你的GPU。估计AMD的没戏。我在Linux环境下，就按照这个项目的README的来设置，自动就用GPU来训练了。我的GPU是 Quadro RTX5000
先看看PyTorch支不支持你的GPU。估计AMD的没戏。我在Linux环境下，就按照这个项目的README的来设置，自动就用GPU来训练了。我的GPU是 Quadro RTX5000
大概完整训练一次的时间是多少呢