From a656be62b671bcf82849c1db86225413c56efbb6 Mon Sep 17 00:00:00 2001
From: Guillermo Rodriguez <guillermo@telos.net>
Date: Sat, 24 Dec 2022 10:54:33 -0300
Subject: [PATCH] Disable torch import on cuda-less runs

---
 skynet/cli.py | 8 ++++++--
 1 file changed, 6 insertions(+), 2 deletions(-)

diff --git a/skynet/cli.py b/skynet/cli.py
index a2985d9..4082a34 100644
--- a/skynet/cli.py
+++ b/skynet/cli.py
@@ -1,4 +1,6 @@
 #!/usr/bin/python
+import importlib.util
+torch_enabled = importlib.util.find_spec('torch') != None
 
 import os
 import json
@@ -10,8 +12,10 @@ import trio
 import click
 import trio_asyncio
 
-from . import utils
-from .dgpu import open_dgpu_node
+if torch_enabled:
+    from . import utils
+    from .dgpu import open_dgpu_node
+
 from .brain import run_skynet
 from .constants import ALGOS