added Util.recover_original_aligned_filename

2019-03-04 10:57:44 +04:00
parent ae4b6e6271
commit 6094e5ead8
5 changed files with 70 additions and 6 deletions
@@ -3,7 +3,9 @@ import json
 import numpy as np
 from .pynvml import *

-tf_min_req_cap = 37 #min req compute capability for tensorflow-gpu==1.11.0
+ 
+
+tf_min_req_cap = int(os.environ.get("force_tf_min_req_cap", 37)) #min req compute capability for tensorflow-gpu==1.11.0
    
 class device:
    backend = None