github/DeepFaceLab
1
0
Fork 0
mirror of https://github.com/iperov/DeepFaceLab.git synced 2025-07-12 16:13:51 -07:00
Code Issues Projects Releases Packages Wiki Activity Actions

upd device.py

Browse source
This commit is contained in:
iperov 2019-03-04 11:17:08 +04:00
parent bdb45357ee
commit c3eea0cf98
1 changed files with 3 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

5
nnlib/device.py
View file

@ -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