安装matplotlib 3.2.2
sudo apt-get install python3.6-dev python3.6-tk tk-dev
sudo apt-get install libfreetype6-dev g++
pip3.6 install matplotlib==3.2.2 --user -i https://pypi.tuna.tsinghua.edu.cn/simple
安装torch、torchvision
sudo apt-get install libopenblas-dev libopenmpi-dev
sudo apt-get --yes install libjpeg-dev zlib1g-dev libpython3-dev libavcodec-dev libavformat-dev libswscale-dev
pip3.6 install torch==1.9.0 torchvision==0.10.0 -i https://pypi.tuna.tsinghua.edu.cn/simple
查看python库部分包版本
pip3.6 list
如果版本不对应或没有的:
pip3.6 uninstall numpy #版本不对应 先卸载后下载
pip3.6 install numpy==1.16.3 -i https://pypi.tuna.tsinghua.edu.cn/simple
pip3.6 install onnxruntime==1.10.0 -i https://pypi.tuna.tsinghua.edu.cn/simple
使用python命令查看是否正常导入包:
python3
import cv2,numpy,torchvision,torch,onnxrumtime