- 1.使用anaconda创建一个环境 (conda create --name maze python=3.10)
- 2.安装必要的包,安装命令在 (command.txt)
-
- 这是必要的包
-
pip install -r requirements.txt
-
- 这是pytorch和cuda的安装
-
pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118
- 直接运行train_dqn.py进行训练
- train_dqn.py 这个文件最下面的有如下代码
if __name__ == '__main__':
main() #训练用的方法
# test_model() #测试用的方法,这里是测试训练好的模型,测试时打开