捉蟲 | mmlab 系列安裝過程遇到的問題合集
阿新 • • 發佈:2021-02-05
(不定時更新…以我遇到問題為主)
- linux
- python3.6
- cuda 10.1
- torch 1.6.0;vision 0.7
mmcv安裝不完全
20200117
事故發生在一開始。
官方教程提供對應版本的mmcv-full下載;但是無法完整獲取。
# latest
pip install mmcv-full -f https://download.openmmlab.com/mmcv/dist/cu101/torch1.6.0/index.html
建議使用這種方式:
git clone https://github.com/open-mmlab/mmcv.git
cd mmcv
# 安裝完整版本
MMCV_WITH_OPS=1 pip install -e .
報錯:No CUDA runtime is found…
20200118
事故發生在mmcv成功後,安裝mmpose和mmdetection期間。
# mmpose
git clone https://github.com/open-mmlab/mmpose.git
cd mmpose
pip install -r requirements.txt # 安裝依賴撲街
python setup.py develop # 正式安裝報錯
# mmdetection
git clone https://github.com/open-mmlab/mmdetection.git
cd mmdetection
pip install -r requirements/build.txt
pip install -v -e . # 注意最後還有一個『.』
# or "python setup.py develop"
直接操作,不仔細看檔案內容安裝的話;
在呼叫測試檔案的時候,會報錯“No CUDA runtime is found”或者“too old”
仔細觀察,可以發現之前 pip install -r requirements.txt
的時候自動給我的torch頂滿了。
如果此時接著使用普通的 pip 重灌對應版本的 torch 依舊會有以上問題。
正確的姿勢應該是:去官網找對應版本,然後重灌。
pip install torch==1.6.0+cu101 torchvision==0.7.0+cu101 -f https://download.pytorch.org/whl/torch_stable.html
之後在python裡輸 torch.cuda.is_available()
驗證一下;只要是 True
這個問題就解決了!!!