second.pytorch環境配置

second.pytorch環境配置記錄


程序運行環境:
運行系統:Ubuntu18.04

使用語言:Python3.6

硬件支持:GTX1660Ti + CUDA10.0+ PyTorch 1.4.0+torchvision0.5+Cmake3.13.2+Anaconda3.5.2

需要注意的點:
下載spconv時注意使用命令 git clone xxx.git --recursive
編譯spconv出錯可能是(pytorch和torchvision版本不匹配)
創建conda虛擬環境
pytorch和torchvision均選擇離線安裝
注意添加second.pytorch環境變量:

export  PYTHONPATH=$PYTHONPATH:/your_second.pytorch_path/ 

一定要安裝numpy1.14.0的版本,過高或過低都會出錯
python create_data.py kitti_data_prep KITTI_DATASET_ROOT(此條命令不加–data_path=)
安裝時可能有其他依賴庫沒有安裝,缺什麼庫安裝什麼即可

還沒有嘗試的點:
目前環境只運行了KITTI 3D detection
沒有使用fp16,沒安裝apex
只嘗試了單GPU訓練

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章