Detectron安裝cuda9.2

一、annaconda安裝虛擬環境

conda create -n detectron python==2.7

二、安裝caffe2 Caffe2官網安裝教程

conda install pytorch-nightly cudatoolkit=9.0 -c pytorch

注意,對於cuda版本爲9.2的,如果直接使用conda install pytorch-nightly -c pytorch命令,則默認安裝的是10.0的版本,會導致後面出錯"CUDA driver version is insufficient for CUDA runtime version"。

三、安裝future、protobuf

pip install matplotlib==2.1 protobuf future pycocotools Cython

四、對照官網安裝

Detectron官網安裝教程

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