(深度学习/深度估计/GLPDepth)
遇到的问题
直接运行pip install -r requirement.txt可能会有如下问题:
-
找不对对应的mmcv版本或者缺少mmcv.runner,安装会报没有对应的wheel
-
torch not compiled with CUDA enabled
问题分析
python、pytorch、cuda(cudatoolkit)、mmcv版本要匹配,CUDA驱动版本要大于需要安装的版本
-
创建虚拟环境,最好创建时就制定python版本:
conda create -n (环境名) python=3.x
-
上pytorch官网查询对应版本:
https://pytorch.org/get-started/previous-versions/
mmcv需降级到1.4,不然后面会报错(缺少mmcv.runner模块,安装会报没有对应的wheel)
所以pytorch版本选择很重要(conda安装是pytorch,pip安装是torch,所以一定要用conda安装!!!)
conda install pytorch==1.8.1 torchvision==0.9.1 torchaudio==0.8.1 cudatoolkit=11.3 -c pytorch -c conda-forge
或conda install pytorch==1.9.0 torchvision==0.10.0 cudatoolkit=11.1 -c pytorch
以下错误是因为用pip安装了torch
解决方案
!按照github上测试环境一比一配置就没问题。即按照测试环境配置,不要直接运行pip install -r requirements.txt