From 4522c9561b31ed45e85e36b06272dd0402c26caa Mon Sep 17 00:00:00 2001 From: jlevine18 Date: Mon, 23 Sep 2019 00:01:31 -0500 Subject: [PATCH] Set device bc I apparently forgot to do that --- data analysis/cudaregress.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/data analysis/cudaregress.py b/data analysis/cudaregress.py index 800c8515..8cdbe141 100644 --- a/data analysis/cudaregress.py +++ b/data analysis/cudaregress.py @@ -38,6 +38,8 @@ __all__ = [ import torch +#set device +device='cuda:0' if torch.cuda.is_available() else 'cpu' #todo: document completely