系统环境中安装的是Python3.5,项目中需要的Python为3.6及以上的环境,所以用conda虚拟环境进行隔离。
conda





项目搭建
进入虚拟环境,安装所需要的包。
sh Miniconda3-py37_4.11.0-Linux-x86_64.sh
source ~/.bashrc
conda config --set auto_activate_base false
#创建虚拟环境
conda create -n pytorch python=3.6
#进入虚拟环境
conda activate pytorch
#从清华源下载安装pytorch
conda config --add channels http://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda install -n pytorch pytorch torchvision cudatoolkit=10.0
#工程需要的库用conda安装会失败,所以用的pip进行安装
#安装insightface,可以用pip安装或者python3.6 -m pip install 来安装
pip install insightface
#安装pymatting
pip install pymatting
#安装onnxruntime,onnxruntime和cuda需要版本对应,cuda10.0对应onnxruntime1.0或1.1
pip install onnxruntime-gpu==1.0
#项目搭建依赖的库用conda就可以
#安装gunicorn
conda install gunicorn
#安装flask
conda install flask
conda install gevent
#gunicorn启动项目
gunicorn -c config.py flask_matting:app
问题
onnxruntime和cuda版本不匹配问题
-
问题
onnxruntime与cuda版本不匹配,可能出现的问题。

本文档记录了在Python3.5系统环境中,如何使用conda创建Python3.6的虚拟环境,并在其中安装PyTorch、ONNXRuntime等库。过程中遇到了ONNXRuntime与CUDA版本不匹配的问题,通过检查CUDA版本并调整ONNXRuntime版本解决了问题。此外,还解决了gunicorn在虚拟环境下不能通过配置文件启动的问题,需修改work_class。同时,对于可能出现的Pillow和Numpy版本错误,给出了具体解决方案。
最低0.47元/天 解锁文章
1309

被折叠的 条评论
为什么被折叠?



