源码编译安装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

posted @ 2020-03-12 18:57  Cheney_1016  阅读(2118)  评论(0编辑  收藏  举报