Need help - Takes forever on newly setup machine

[thiago]

Cool, now you could test if pytorch is using your GPU.

Check it out:

How to check your pytorch / keras is using the GPU?

As we work on setting up our environments, I found this quite useful: To check that torch is using a GPU: In [1]: import torch In [2]: torch.cuda.current_device() Out[2]: 0 In [3]: torch.cuda.device(0) Out[3]: <torch.cuda.device at 0x7f2132913c50> In [4]: torch.cuda.device_count() Out[4]: 1 In [5]: torch.cuda.get_device_name(0) Out[5]: 'Tesla K80' To check that keras is using a GPU: import tensorflow as tf tf.Session(config=tf.ConfigProto(log_device_placement=True)) and check the jupyte…