强化学习包PARL1.3.1，使用cpu飞桨
0
收藏
在使用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.
此时会终止训练，然后提示安装gpu版本得paddle。我们只需要修改一下源码就可以继续运行，且后续安装gpu版本仍然可以使用。
首先找到python安装路径，然后寻找Lib/site-packages/parl/utils
找到machine_info.py文件，然后在105行后， 也就是is_gpu_available这个函数中添加ret=False。
具体看图片
收藏
点赞
0
个赞
请登录后评论
TOP
切换版块
不错的，加油
赞