环境:
ubuntu16.04
nvidia460,cuda11,cudnn8
python3.7.10:
absl-py 0.12.0
astor 0.8.1
cached-property 1.5.2
certifi 2020.12.5
cycler 0.10.0
gast 0.2.2
google-pasta 0.2.0
grpcio 1.36.1
h5py 2.10.0
importlib-metadata 3.7.3
Keras 2.1.5
Keras-Applications 1.0.8
Keras-Preprocessing 1.1.2
kiwisolver 1.3.1
lxml 4.6.3
Markdown 3.3.4
matplotlib 3.4.0
numpy 1.20.1
opt-einsum 3.3.0
Pillow 8.1.2
pip 21.0