编译选项

You 2023-12-31 00:52:19  64926

分类专栏：资讯

译

编译重头戏来了，编译过程中大家可能会遇到各种各样的问题，但是其实只要我们将环境准备妥当，大部分都可以一次性编译好的：

确保你的cuda和cudnn安装正确，环境变量都设置正确
确保你的python环境纯净，最好使用anaconda创建一个环境，安装好pytorch需要的依赖包。 https://github.com/pytorch/pytorch#from-source
确保你的C++编译器的版本不要太低，最好4.9以及以上

编译选项

python的安装方式并不是单独利用Cmake进行构建的，而是结合了python端的setuptools搭配cmake进行构建，pytorch的项目还是比较庞大的，所以编译代码也是老长，我们主要看看编译过程中的环境变量即可：

# Environment variables you are probably interested in:
#
#   DEBUG
#     build with -O0 and -g (debug symbols)
#
#   REL_WITH_DEB_INFO
#     build with optimizations and -g (debug symbols)
#
#   MAX_JOBS
#     maximum number of compile jobs we should use to compile your code
#
#   NO_CUDA
#     disables CUDA build
#   
#   ....
#   ....
#   
# Environment variables for feature toggles:
#
#   NO_CUDNN
#     disables the cuDNN build
#
#   NO_FBGEMM
#     disables the FBGEMM build
#
#   NO_TEST
#     disables the test build
#
#   NO_MIOPEN
#     disables the MIOpen build

这些编译变量根据我们的需要在执行python setup.py install使用，如果你不想编译CUDA，则NO_CUDA=1 python setup.py install.

执行以上语句我们就可以进行编译了。

网站声明：如果转载，请联系本站管理员。否则一切后果自行承担。

本文链接：https://www.xckfsq.com/news/show.html?id=33090

赞同 0

评论 0 条

编译选项

译

编译选项

相关文章

关注我们