ipykernel 安装
pip install ipykernel
pip install ipywidgets
pip install --upgrade jupyter notebook
适配的torch参数
matplotlib==3.4.2
numpy==1.21.6
prettytable==2.2.1
ptflops==0.6.9
scipy==1.7.3
statsmodels==0.13.5
torch==2.0.1
torchvision==0.15.2
tqdm==4.62.3