About using the GPU

[luxiaojuntokyo]

When I run: python train.py --policy sarl
Every time there will be: INFO: Using device: cpu

I wonder how can I use GPU to speed up the training?

My server's setting, GPU: GTX 1080Ti × 3 (VRAM: 11GB × 3)

Thanks a lot.

[YuejiangLIU]

Please try and add the '--gpu' argument in your command

[ChanganVR]

@luxiaojuntokyo the code runs faster on CPU because the speed bottleneck is on CPU-intensive operations. Feel free to explore and optimize the code such that you could use GPU to speed it up.

[luxiaojuntokyo]

@luxiaojuntokyo the code runs faster on CPU because the speed bottleneck is on CPU-intensive operations. Feel free to explore and optimize the code such that you could use GPU to speed it up.

Changan, thank you for your reply. If I get any result, I will comment here.

[luxiaojuntokyo]

Please try and add the '--gpu' argument in your command

LIU Yuejiang, thank you for your reply.