Bug Fixes: Improved Multi-GPU Support in Windows System and Optimized the Version Numbers of Some Modules.

This commit is contained in:
YuMo 2023-04-22 16:48:29 +08:00
parent a8c039da93
commit e40201f15f
3 changed files with 27 additions and 25 deletions

View file

@ -68,6 +68,7 @@ class Devices(object):
return None
def get_devices_from_index_list(self, idx_list):
if not isinstance(idx_list, list):
idx_list = [int(idx.strip()) for idx in idx_list.split(',')]
result = []
for device in self.devices:

View file

@ -1,12 +1,13 @@
tqdm
numpy==1.19.5
numexpr
h5py==2.10.0
opencv-python==4.1.2.30
ffmpeg-python==0.1.17
scikit-image==0.14.2
scipy==1.4.1
colorama
tensorflow-gpu==2.4.0
cython==0.29.25
ffmpeg-python==0.1.17
h5py==2.10.0
numpy==1.20.1
numexpr
opencv-python==4.1.2.30
pyqt5
scikit-image==0.18.3
scipy==1.10.1
tensorflow-gpu==2.4.1
tf2onnx==1.9.3
tqdm