本人python=3.8
,今天装环境弄了很久,发现使用conda install
无法同时安装 pytorch和opencv
解决方法,来自https://github.com/pytorch/vision/issues/3207
pytorch正常安装
conda install pytorch torchvision torchaudio cudatoolkit=10.2 -c pytorch
opencv使用pip安装
pip install opencv-contrib-python
注意:如果你后续还需要安装基于opencv的albumentations做数据增强,也使用pip安装
pip install -U albumentations
------------------------分割线------------------------
------------------------以下是之前未能解决的时候写的内容,可以忽视------------------------
conda install pytorch torchvision torchaudio cudatoolkit=10.2 -c pytorch
conda install -c conda-forge opencv
pip
行不行不清楚,据说也不行。
然后去google发现是torchvision的问题,原帖子如下
https://github.com/pytorch/vision/issues/4076
按别人的回答:
目前torchvision的版本是torchvision=0.10
,需要ffmpeg>=4.2
;
而opencv>=4.2
才支持ffmpeg>=4.2
;
但是opencv>=4.2
需要jpge>=9d
;
然而torchvision=0.10
又需要jpeg<=9b
所有都满足,只能将torchvision的版本降低至torchvision=0.2.2
conda install -c pytorch torchvision=0.2.2
目前没有有效的解决方案,如果需要用到torchvision最新的功能,你可能需要在它与opencv中选一个。