-
根据官网指示安装
pytorch安装指南:https://github.com/pytorch/pytorch
conda 安装对应的包: https://anaconda.org/anaconda/ (这个网站可以搜索包的源)
-
如果按照官网提供的export cmake_path方式不成功,推荐在~/.bashrc中添加cmake的路径
eg:export CMAKE_PREFIX_PATH="/home/jzc/miniconda3/bin:$CMAKE_PREFIX_PATH"
-
如果是开发模式, 使用build编译安装 python setup.py build develop
-
如果要编译debug模式,使用 Debug=1 python setup.py install
这种方式很有可能无法改变原始设置,编译的时候还是release. 如果出现这种情况,请强制设置debug环境变量, 在pytorch/tools/setup_helpers/env.py中添加 os.environ['DEBUG']='1'
-
NO_CUDA=1 python setup.py install 使用这种方式设置环境变量,可能造成编译的时候无法识别设置好的环境变量
因此,保险点的方法是在CMakelist.txt中设置环境变量。
-
编译的时候,可能会出现 can't find -lxxxx 找不到对应的库。
如果在anaconda的lib中找到了该库,但是gcc/gxx搜索不到,那么进入到 gcc/gxx的安装目录(x86_64-linux-gnu/) 建立软链接,将已有的库文件链接到该目录下
本站文章如无特殊说明,均为本站原创,如若转载,请注明出处:源码编译安装pytorch debug版本 - Python技术站