From c3eea0cf9813a5fdd624718c0817e2011777d6c3 Mon Sep 17 00:00:00 2001 From: iperov Date: Mon, 4 Mar 2019 11:17:08 +0400 Subject: [PATCH] upd device.py --- nnlib/device.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/nnlib/device.py b/nnlib/device.py index 0fac79d..27e973a 100644 --- a/nnlib/device.py +++ b/nnlib/device.py @@ -4,8 +4,9 @@ import numpy as np from .pynvml import * - -tf_min_req_cap = int(os.environ.get("force_tf_min_req_cap", 37)) #min req compute capability for tensorflow-gpu==1.11.0 +#you can force_tf_min_req_cap 35, if your DFL is built for tf==1.5.0 +#the reason why we cannot check tensorflow.version is it requires import tensorflow +tf_min_req_cap = int(os.environ.get("force_tf_min_req_cap", 37)) class device: backend = None