强化学习包PARL1.3.1,使用cpu飞桨报错

在使用CPU版本得PaddlePaddle1.8.1和强化学习包parl1.3.1版本时,进行训练会报错

Found non-empty CUDA_VISIBLE_DEVICES. But PARL found that Paddle was not complied with CUDA, which may cause issues.
if you want to use GPU, please try to install GPU version PaddlePaddle by: pip install paddlepaddle-gpu If you only have CPU, please change CUDAPlace(0) to be CPUPlace().

此时会终止训练,然后提示安装gpu版本得paddle。我们只需要修改一下源码就可以继续运行,且后续安装gpu版本仍然可以使用。

解决方法:
此类问题主要是显卡驱动可以让程序检测出有显卡,所以在运行的时候自动选择了GPU版本的paddlepaddle。我们需要修改一下parl源码。

首先找到python安装路径,然后寻找Lib/site-packages/parl/utils
找到machine_info.py文件,然后在105行后, 也就是is_gpu_available这个函数中添加ret=False。

具体看图片
强化学习包PARL1.3.1,使用cpu飞桨报错_第1张图片

你可能感兴趣的:(PARL,RL,深度学习)