windows上通过cmake-gui生成pytorch工程

Posted fengbingchun

tags:

篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了windows上通过cmake-gui生成pytorch工程相关的知识,希望对你有一定的参考价值。

      在Windows下通过cmake-gui.exe生成不带cuda的Torch.sln工程操作步骤:PyTorch版本使用1.8.1。首先可以先通过打开cmake-gui.exe,指定pytorch源代码目录和生成vs2017工程的位置;然后点击Configure,选中”Visual Studio 15 2017 Win64”,点击Finish,会报一堆CMake Error,然后逐个解决处理,如下图所示:然后反复的点击Configure,直到Configuing done;反复的点击Generate,直到Gererating done

       1. 从github上下载第三方库,解压缩后拷贝到pytorch/third_party对应目录下:

       (1). protobuf:从github/protocolbuffers上下载protobuf-all-3.15.0.zip;

       (2). pthreadpool:从github/Maratyszcza上下载pthreadpool-master.zip;

       (3). FXdiv:从github/Maratyszcza上下载FXdiv-master.zip;

       (4). cpuinfo:从github/pytorch上下载cpuinfo-master.zip;

       (5). XNNPACK:从github/google上下载XNNPACK-master.zip;

       (6). FP16:从github/Maratyszcza上下载FP16-master.zip;

       (7). psimd:从github/Maratyszcza上下载psimd-master.zip;

       (8). fbgemm:从github/pytorch上下载FBGEMM-main.zip;

       (9). asmjit:从github/asmjit上下载asmjit-master.zip,将其拷贝到fbgemm/third_party/asmjit目录下;

       (10). onnx:从github/onnx上下载onnx-1.8.0.zip;

       (11). foxi:从github/houseroad上下载foxi-master.zip;

       (12). fmt:从github/fmtlib上下载fmt-7.1.3.zip;

       2. 处理CMake Error “cmake/Codegen.cmake:205 (message): Failed to get generated_cpp list”:

       (1). pytorch 1.8.1对应的python版本为3.8.8,在本机安装此版本的python;

       (2). pip install typing-extensions

       (3). pip install pyyaml

       到此为止,应该可以Configuing done了。

       3. 调整XNNPACK后应该可以Generating done:

       (1). “Cannot find source file: src/x8-transpose/gen/16x16-sse2.c”,将CMakeLists.txt中的第3953行注释掉;

       (2). “Cannot find source file: src/x16-transpose/gen/8x8-sse2.c”,将CMakeLists.txt中的第3959行注释掉;

       (3). “Cannot find source file: src/x32-transpose/gen/4x4-sse2.c”,将CMakeLists.txt中的第3960行注释掉;

       (4). “Cannot find source file: src/x64-transpose/gen/2x2-sse2.c”,将CMakeLists.txt中的第3966行注释掉;

       Configuing done和Gererating done的结果如下图所示:

       4.  打开PyTorch_VS2017下的Torch.sln工程:结果如下图所示

     通过vs code打开pytorch源码看c++ code时有时不是很方便,这里直接通过Torch.sln查看有时会方便很多。虽然可以生成Torch.sln工程,但是并不能全部编译通过,编译时还会报一些error,如fbgemm,待后面用到时再处理。

      GitHubhttps://github.com/fengbingchun/PyTorch_Test

以上是关于windows上通过cmake-gui生成pytorch工程的主要内容,如果未能解决你的问题,请参考以下文章

嵌入式linux开发,opencv移植,opencv-2.4.9移植,cmake-gui

windows下编译google-protocolbuf在Qt中使用

哪个版本的 Insight 工具包支持 libxml2?

centos6.5安装cmake-gui

linux下cmake-gui的安装

cmake编译Qt工程