LmingXie
commented
1 year ago 我的问题是使用了错误tensorflow和tensorflow-gpu版本，我使用cuda 11.4，安装tensorflow 2.3.0和tensorflow-gpu 2.3.0 可以运行项目，输入
# 进入python
import tensorflow as tf
print(tf.test.is_gpu_available())测试也能通过，但tensorflow-gpu 2.3.0适配CUDA版本是10.1，11.4选用2.7.4版本后yolov3可以正常使用gpu，至于为什么GPU可用性测试能通过未找到原因。
Package Version
---------------------------- ----------
absl-py 0.15.0
aiohttp 3.8.3
aiosignal 1.2.0
altgraph 0.17.3
astor 0.8.1
astunparse 1.6.3
async-timeout 4.0.2
asynctest 0.13.0
attrs 22.1.0
blinker 1.4
brotlipy 0.7.0
cached-property 1.5.2
cachetools 4.2.4
certifi 2022.12.7
cffi 1.15.1
chardet 5.1.0
charset-normalizer 2.1.0
clang 5.0
click 8.1.3
colorama 0.4.6
cryptography 39.0.1
cycler 0.11.0
flatbuffers 1.12
flit_core 3.6.0
fonttools 4.38.0
frozenlist 1.3.3
gast 0.4.0
google-auth 2.18.0
google-auth-oauthlib 0.4.6
google-pasta 0.2.0
grpcio 1.54.2
h5py 3.8.0
idna 3.4
importlib-metadata 6.6.0
keras 2.7.0
Keras-Preprocessing 1.1.2
kiwisolver 1.4.4
labelImg 1.8.6
libclang 16.0.0
loguru 0.7.0
lxml 4.9.1
Markdown 3.4.3
MarkupSafe 2.1.2
matplotlib 3.5.3
mkl-fft 1.3.1
mkl-random 1.2.2
mkl-service 2.4.0
multidict 6.0.2
numpy 1.19.2
oauthlib 3.2.2
opencv-python 4.7.0.72
opt-einsum 3.3.0
packaging 23.1
pefile 2023.2.7
Pillow 9.5.0
pip 23.1.2
protobuf 3.19.6
pyasn1 0.5.0
pyasn1-modules 0.3.0
pycparser 2.21
pyinstaller 5.10.1
pyinstaller-hooks-contrib 2023.2
PyJWT 2.4.0
pyOpenSSL 23.0.0
pyparsing 3.0.9
PyQt5 5.15.9
pyqt5-plugins 5.15.9.2.3
PyQt5-Qt5 5.15.2
pyqt5-tools 5.15.9.3.3
PySocks 1.7.1
python-dateutil 2.8.2
python-dotenv 0.21.1
pywin32-ctypes 0.2.0
qt5-applications 5.15.2.2.3
qt5-tools 5.15.2.1.3
requests 2.30.0
requests-oauthlib 1.3.1
rsa 4.9
scipy 1.7.3
setuptools 65.6.3
six 1.15.0
spicy 0.16.0
tensorboard 2.11.2
tensorboard-data-server 0.6.1
tensorboard-plugin-wit 1.8.1
tensorflow 2.7.4
tensorflow-estimator 2.7.0
tensorflow-gpu 2.7.4
tensorflow-gpu-estimator 2.3.0
tensorflow-io-gcs-filesystem 0.31.0
termcolor 1.1.0
torch 1.12.1
torchaudio 0.12.1
torchvision 0.13.1
tqdm 4.65.0
typing-extensions 3.7.4.3
urllib3 1.26.15
Werkzeug 2.2.3
wheel 0.35.1
win-inet-pton 1.1.0
win32-setctime 1.1.0
wincertstore 0.2
wrapt 1.12.1
xyolo 0.1.6
yarl 1.8.1
zipp 3.15.0
版本参考：https://tensorflow.google.cn/install/source_windows?hl=zh-cn#gpu
我在运行这段代码时发现，gpu利用率很低，基本保持在10%以下，cpu全程保持在100%，大佬又遇到这个问题吗？
然后导致一直到检测，运行一个多小时仍然无法得出，目标检测模型。使用的是master分支上的图片数据。