Open ChenyangRan opened 3 years ago
Hi, I have read the doc and config. But it seems that no flag to set the use_gpu. I have test the sac_pytorch and it seems the gpu is not used. The python is 3.6 and pytorch is 1.3.1 Could you please tell how to set?
The implementation doesn't support GPUs. Check out https://github.com/PyTorchLightning/pytorch-lightning-bolts
Hi, I have read the doc and config. But it seems that no flag to set the use_gpu. I have test the sac_pytorch and it seems the gpu is not used. The python is 3.6 and pytorch is 1.3.1 Could you please tell how to set?