安装必要的库

pip install numpy
pip install pillow
pip install opencv-python
pip install pywin32
pip install pandas
pip install gym

pip install tensorflow
pip install tensorflow-gpu
 
pip install numpy opencv-python pillow pywin32 gym pandas
pip install tensorflow

pip install tensorflow -i https://pypi.tuna.tsinghua.edu.cn/simple # tensorflow装不了的话用清华源会快一点

# 如果需要ocr来定位视角就安装
pip install pyautogui pytesseract
https://github.com/UB-Mannheim/tesseract/wiki

cuda 安装

选择系统适合的cuda版本: https://developer.nvidia.com/cuda-toolkit-archive

C:\Users\Administrator>nvidia-smi
Sat Aug 24 22:21:03 2024
+-----------------------------------------------------------------------------------------+
| NVIDIA-SMI 560.94                 Driver Version: 560.94         CUDA Version: 12.6     |

开始训练

python training.py

模型试跑

python testing.py

部分代码来自以下仓库，感谢开源

• https://github.com/analoganddigital/DQN_play_sekiro
• https://github.com/Sentdex/pygta5

更通用的框架

• https://github.com/Tencent/GameAISDK