libtorch編譯時只使用單核cpu的問題

在ubuntu18.04虛擬機下面,從源碼編譯libtorch,發現編譯速度超級慢,一看cpu只使用了一個核,應該是配置的時候參數有點問題,找了一下發現,應該是有個參數讀出來的不太對,可能是因爲虛擬機配置讀取的問題,文件路徑是pytorch/scripts/build_local.sh,最後一行的參數修改下,如下圖 估計是這個CAFFE_MAKE_NCPUS的數據錯了,如下,直接把變量改成4或者8
相關文章
相關標籤/搜索