源碼編譯安裝pytorch debug版本


  1. 根據官網指示安裝

    pytorch安裝指南:https://github.com/pytorch/pytorch

    conda 安裝對應的包: https://anaconda.org/anaconda/ (這個網站可以搜索包的源)

  2. 如果按照官網提供的export cmake_path方式不成功,推薦在~/.bashrc中添加cmake的路徑

    eg:export CMAKE_PREFIX_PATH="/home/jzc/miniconda3/bin:$CMAKE_PREFIX_PATH"

  3. 如果是開發模式, 使用build編譯安裝 python setup.py build develop

  4. 如果要編譯debug模式,使用 Debug=1 python setup.py install

    這種方式很有可能無法改變原始設置,編譯的時候還是release. 如果出現這種情況,請強制設置debug環境變量, 在pytorch/tools/setup_helpers/env.py中添加 os.environ['DEBUG']='1'

  5. NO_CUDA=1 python setup.py install 使用這種方式設置環境變量,可能造成編譯的時候無法識別設置好的環境變量

    因此,保險點的方法是在CMakelist.txt中設置環境變量。

  6. 編譯的時候,可能會出現 can't find -lxxxx 找不到對應的庫。

    如果在anaconda的lib中找到了該庫,但是gcc/gxx搜索不到,那么進入到 gcc/gxx的安裝目錄(x86_64-linux-gnu/) 建立軟鏈接,將已有的庫文件鏈接到該目錄下

    https://blog.csdn.net/qq_39436605/article/details/80893885


免責聲明!

本站轉載的文章為個人學習借鑒使用,本站對版權不負任何法律責任。如果侵犯了您的隱私權益,請聯系本站郵箱yoyou2525@163.com刪除。



 
粵ICP備18138465號   © 2018-2025 CODEPRJ.COM