1. 程式人生 > 其它 >捉蟲 | mmlab 系列安裝過程遇到的問題合集

捉蟲 | mmlab 系列安裝過程遇到的問題合集

技術標籤:HPEdebug

(不定時更新…以我遇到問題為主)

  • 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 這個問題就解決了!!!