pytorch原始碼編譯遇到的問題

2021-08-28 10:04:34 字數 589 閱讀 5503

1. 編譯到大概33%處,出現failed to run 'bash ../tools/build_pytorch_libs.sh --use-cuda --use-nnpack nccl caffe2 libshm gloo c10d thd'

解決:最新的pytorch原始碼編譯要求cudnn>5,公升級cudnn到6.0以上的版本即可。

不需要更改/usr/local/cuda下邊的cudnn庫,參考以下方法,系統中可安裝多個cuda版本和cudnn版本的方法。

將自己的cudnn目錄加到ld_library_path中即可,其他需要特定cudnn版本的軟體包編譯也可通過此法解決。

2. 以上貌似後續還有問題,沒能解決。在74%處還有問題,需要設定其他幾個環境變數。

cudnn_library和cudnn_include_dir,分別設定到自定義cudnn版本處的lib64資料夾和include資料夾。

在終端中設定好像還是有問題,可能是設定方法不對,最後看到編譯指令碼是py檔案,於是嘗試在pycharm中執行,run/debug configuration視窗可以設定環境變數。

此外,還可以設定max_jobs環境變數,多執行緒編譯,來加快編譯速度。

原始碼編譯pytorch 1 0 遇到的問題

一 問題1pytorch stable third party ideep mkl dnn src cpu ref rnn.cpp 在成員函式 void mkldnn impl cpu ref rnn common t pack weights int,int,int,int,int,int,int...

Pytorch原始碼注釋

field類為可以由張量表示的常見文字處理資料型別建模。它包含乙個vocab物件,用於定義字段元素的可能值集及其對應的數字表示。field物件還包含與資料型別應如何數位化有關的其他引數,例如標記化方法和應生成的tensor型別。如果在資料集中的兩列之間共享字段 例如,qa資料集中的問題和答案 則它們...

更新自己編譯android原始碼遇到的問題

1 bin bash bison 未找到命令 安裝bison sudo apt get install bison 2 借鑑前輩經驗先安裝了一些工具包參考部落格 前輩的部落格 寫的好整齊 總結一下安裝 sudo apt get install gnupg flex bison gperf libsd...