1. git clone --recursive https://github.com/apache/incubator-mxnet mxnet
2. 在mxnet下创建目录build2019和install
3. 运行CMake-GUI , 选择源文件目录为mxnet,构建目录为build2019, 构建目标为vs2019/x64, -T选项为cuda=/path/to/cuda
4. 点击configure, 报告cuda compiler找不到, 填写CMAKE_CUDA_COMPILER为nvcc.exe的全路径,再次点击configure
5. 报错找不到vs-integration的文件,故拷贝CUDA\v10.1\extras\visual_studio_integration\MSBuildExtensions下面的所有文件到
C:\Program Files (x86)\Microsoft Visual Studio\2019\BuildTools\MSBuild\Microsoft\VC\v160\BuildCustomizations
6. 点击configure,这次CUDA不再报错。 填写OpenBLAS, OPENCV_DIR等依赖项路径
7. configure 和 generate
网友评论