只是需要其中的几个文件，所以只想简单编译，简单编译也不好配置。
下载源码：
git clone https://github.com/pytorch/pytorch
git checkout v2.6.0-rc9
git submodule update --init --recursive
编译源码：
可以用cmake gui打开，但是编译错误会比较多。这里使用python命令。
python setup.py build
如果编译出错，修改配置之后，需要先删除build目录，否则可能报错如下：
pytorch\tools\setup_helpers\cmake_utils.py", line 57, in get_cmake_cache_variables_from_file
for i, line in enumerate(cmake_cache_file, 1):
UnicodeDecodeError: 'gbk' codec can't decode byte 0xb6 in position 77: illegal multibyte sequence
如果需要禁用某些特性，可以在命令行先设置环境变量，以此开启或关闭某些模块的编译。例如：
set USE_CUDA=0
set USE_CUDNN=0
set USE_OPENMP=0
set USE_TENSORPIPE=0
set USE_CUSPARSELT=0
set USE_CUDSS=0
set USE_CUFILE=0
set USE_FBGEMM=0
set USE_KINETO=0
set BUILD_TEST=0
set USE_NUMPY=0
set USE_MKLDNN=0
set USE_ITT=0
set USE_STATIC_MKL=0
set USE_NNPACK=0
set USE_DISTRIBUTED=0
set USE_GLOO=0
set USE_MPI=0
set USE_SYSTEM_NCCL=0
set USE_FLASH_ATTENTION=0
set USE_MEM_EFF_ATTENTION=0
set USE_ROCM_KERNEL_ASSERT=0
以上禁用了大部分特性。因为只需要其中的极少数功能，所以此处禁用掉这些。
如需要指定Visual Studio版本，可以打开相应版本的VS编译命令行环境，再执行以上命令。
没用上这个。dll太大了，改成lib由提示lib库过大 无法生成。精简后生成lib了，但是一些脚本代码没法执行，只能放弃。
评论0
暂时没有评论