alu/tra-analysis
1
0
Fork 0
mirror of https://github.com/titanscouting/tra-analysis.git synced 2024-12-27 18:09:08 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix: only import cupy if cuda available

Browse Source 
Signed-off-by: Dev Singh <dev@devksingh.com>
This commit is contained in:
Dev Singh 2020-09-22 19:08:23 +00:00
parent ab40e66a2c
commit 9aa886293f
1 changed files with 4 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
analysis-master/tra_analysis/regression.py
View File

@ -48,13 +48,16 @@ __all__ = [
] ]
import torch import torch
import cupy as cp
import numpy as np import numpy as np
global device global device
device = "cuda:0" if torch.torch.cuda.is_available() else "cpu" device = "cuda:0" if torch.torch.cuda.is_available() else "cpu"
if device !== "cpu":
import cupy as cp
#todo: document completely #todo: document completely
def set_device(self, new_device): def set_device(self, new_device):