diff --git a/src/Car-Racing/README.md b/src/Car-Racing/README.md index 2dc40a2964..4c6daf82f4 100644 --- a/src/Car-Racing/README.md +++ b/src/Car-Racing/README.md @@ -4,6 +4,8 @@ 使用方法,把权重模型文件放在training目录下面的saved_models目录下,eval即可自己调用 2.代码路径需要改成本地的,需要注意 3.下图是运行入口训练代码三个小时的奖励曲线图 -4.configs里的文件是训练模型的超参数配置文件,可以改参数 ![rewardline](https://raw.githubusercontent.com/GODDDDD22311/assets/main/rewardline.PNG) - +4.configs里的文件是训练模型的超参数配置文件,可以改参数 +5.注意需要安装cu121(cuda)依赖,即需要启动gpu进行训练,cpu训练可能需要以天为单位进行训练,可以新建终端用gpucheck.txt里的指令查看显存+利用率 +下图为指令实图 +![gpucheck](https://raw.githubusercontent.com/GODDDDD22311/assets/main/gpucheck.gif) diff --git a/src/Car-Racing/gpucheck.txt b/src/Car-Racing/gpucheck.txt new file mode 100644 index 0000000000..e493a690c6 --- /dev/null +++ b/src/Car-Racing/gpucheck.txt @@ -0,0 +1 @@ +nvidia-smi --query-gpu=timestamp,memory.used,memory.total,utilization.gpu --format=csv -l 1