本文介紹Ubuntu下配置車道線檢測算法LaneATT代碼運行環境,步驟如下。
1. 從LaneATT官方代碼倉庫下載源碼。也可git直接拉取。
2. 安裝Anaconda或miniconda 。
參考:https://docs.anaconda.com/miniconda/
3. 創建conda虛擬環境LaneATT(環境名可自定義)
conda create -n LaneATT python=3.8
4. 進入虛擬環境
conda activate LaneATT
5. 安裝Pytorch 1.6.0、torchvision 0.7.0、cuda 10.2
conda install pytorch==1.6.0 torchvision==0.7.0 cudatoolkit=10.2 -c pytorch
6. 在工作目錄下創建LaneATT文件夾(名字可自定義),將LaneATT代碼上傳到該文件夾內。
7. 在命令行中切換工作目錄到代碼文件夾
8. 安裝依賴庫
pip install -r requirements.txt
9. 修改setuptools版本,否則可能報錯
pip install setuptools==58.2.0
10. 切換到lib/nms路徑
cd lib/nms
11.?運行setup.py
python setup.py
12. 若安裝過程出現gcc報錯:“gcc: error trying to exec 'cc1plus': execvp: 沒有那個文件或目錄,”則在虛擬環境中重新安裝gcc和g++,確保兩者版本一致(該錯誤就是gcc和g++版本不一致導致的),以下以從清華鏡像源安裝以8.5.0版本為例。
conda install -c https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main gcc_impl_linux-64=8.5.0
conda install -c https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main gxx_impl_linux-64=8.5.0
13. 切換回工作目錄
cd –
14. 運行main.py打印help信息,若出現下圖信息,則說明配置成功。
python main.py –help
15. 若出現以下numpy報錯,則將numpy版本降級到1.20以下。
pip install numpy==1.18.5 -i https://pypi.tuna.tsinghua.edu.cn/simple
16. 若出現以下報錯,則將protobuf降級到3.20以下。
pip install protobuf==3.19.0 -i https://pypi.tuna.tsinghua.edu.cn/simple --trusted-host pypi.tuna.tsinghua.edu.cn