Set device in notebook

[andrewzhang505]

Set device to cpu in notebook if cuda not available for #261

[codecov-commenter]

Codecov Report

Base: 79.49% // Head: 79.52% // Increases project coverage by +0.02% 🎉

Coverage data is based on head (528a44d) compared to base (7509cc4).
Patch has no changes to coverable lines.

Additional details and impacted files

@@            Coverage Diff             @@
##           master     #262      +/-   ##
==========================================
+ Coverage   79.49%   79.52%   +0.02%     
==========================================
  Files          96       96              
  Lines        7486     7486              
==========================================
+ Hits         5951     5953       +2     
+ Misses       1535     1533       -2     

Impacted Files	Coverage Δ
sample_factory/algo/learning/learner.py	87.99% <0.00%> (-0.16%)	⬇️
sample_factory/pbt/population_based_training.py	85.00% <0.00%> (+0.41%)	⬆️
sample_factory/algo/utils/make_env.py	95.81% <0.00%> (+0.83%)	⬆️

Help us with your feedback. Take ten seconds to tell us how you rate us. Have a feature suggestion? Share it here.

☔ View full report at Codecov.
📢 Do you have feedback about the report comment? Let us know in this issue.

[alex-petrenko]

Thank you!!