Closed ma3252788 closed 4 years ago
You can run it on just CPU to get started. Add
import os
os.environ['CUDA_VISIBLE_DEVICES'] = ''
on top of the python script to bypass Nvidia GPUs.
You can run it on just CPU to get started. Add
import os os.environ['CUDA_VISIBLE_DEVICES'] = ''
on top of the python script to bypass Nvidia GPUs.
Does the CPU and GPU training time have a big difference? How long does it take to train if you use a CPU?
https://github.com/hongzimao/pensieve/issues/82 The speed depends on your hardware. Since the model is not too large, CPU and GPU performed similarly in our experiment.
Can the program be trained using the GPU? I used the GPU. As a result, each agent needs to allocate memory. The result shows that the GPU memory is insufficient. Can anyone help me? Thank you