windows pytorch caffe2 安装

最新推荐文章于 2023-07-16 19:20:02 发布

阿达在学DL

最新推荐文章于 2023-07-16 19:20:02 发布

阅读量1.1k

点赞数

CC 4.0 BY-SA版权

文章标签： caffe2 windwos pytorch

本文链接：https://blog.youkuaiyun.com/qq_42707158/article/details/81063778

本文档详细介绍了如何在Windows操作系统中安装PyTorch及其依赖库Caffe2，通过运行build_windows.bat脚本来完成配置和构建过程。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

下载 Visual Studio Community Edition 2017

下载Python 2.7.14 (x86-64) 和 Microsoft Visual C++ Compiler for Python 2.7.

把 C:\Python27 and C:\Python27\scripts 加入到windows Path变量中。
下载下列依赖包

pip install future ^
            hypothesis ^
            numpy ^
            protobuf ^
            six

pip install flask ^
            glog ^
            graphviz ^
            jupyter ^
            matplotlib ^
            pydot python-nvd3 ^
            pyyaml ^
            requests ^
            scikit-image ^
            scipy ^
            setuptools ^
            tornado

clone pytorch源码到本地。

git clone --recursive https://github.com/pytorch/pytorch.git

编辑 ‘\scripts\build_windows.bat’
找到其中的 set BUILD_PYTHON 设置为 ON

运行Developer Command Prompt for VS 2017,

找到pytorch中\scripts文件夹，先在命令行中运行set CMAKE_GENERATOR="Visual Studio 15 2017 Win64" 将CAMKE编译器换成2017版本然后在命令行中输入

call build_windows.bat

最后在环境变量中加入PYTHONPATH这个环境变量
路径设置为pytorch\build 所在路径。大功告成