Win11编译cpu版本的pytorch源码

本文介绍如何从源码编译PyTorch,并详细记录了在Windows 11环境下,使用Anaconda3和Visual Studio 2019进行编译的过程。包括创建虚拟环境、配置编译环境及解决常见错误。

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

1、下载源码和依赖库

      首先,在github上下载指定版本的pytorch源码,地址如下:https://github.com/pytorch/pytorch,因为在pytorch源码中有一些依赖库是直接给出的超链接,因此都需要直接离线下载依赖库。pytorch依赖库均下载到third_parth目录中。

2、编译pytorch源码

     编译环境win11,anaconda3。

2.1 创建虚拟环境

首先,利用conda创建虚拟环境:

conda create -n pytorch1.10 python==3.7

进入虚拟环境中:

conda activate pytorch1.10

查看是否创建成功:

2.2  编译

首先,进入pytorch源码目录下,进入终端,之后输入cmd,因为是cpu版本直接设置USE_CUDA=0,设置REL_WITH_DEB_INFO=1,否则windows环境下无法进入C++ debug模式。

最后,执行python setup.py develop开始执行编译。

编译过程中的出现的error:

1.could not find any instance of Visual Studio

   报这个错是因为windows上没有适配到MSVC编译器,我们需要在path中添加这MSCV,具体操作可以参考这个博客,Visual Studio 2019 配置 MSVC 环境变量,使用命令行编译。 - 简书 (jianshu.com)

 之后,输入自己所在的dat文件:

"C:\Program Files\Microsoft Visual Studio\2022\Community\VC\Auxiliary\Build\vcvarsall.bat" x64 -vcvars_ver=14.3

 

 

要安装CPU版本PyTorch,你可以按照以下步骤进行操作: 1. 首先,确保你的电脑上已经安装了Python 3.5或Python 3.6版本,因为Windows下的PyTorch仅支持这两个版本。如果你还没有安装Python,你可以从官方网站下载并安装。 2. 接下来,你可以通过conda、pip或源码编译三种方式来安装PyTorch。如果你使用conda,可以在Anaconda Prompt中输入以下命令来安装PyTorch: ``` conda install pytorch-cpu torchvision-cpu -c pytorch ``` 如果你使用pip,可以在命令提示符中输入以下命令来安装PyTorch: ``` pip install torch==1.9.0+cpu torchvision==0.10.0+cpu torchaudio==0.9.0 -f https://download.pytorch.org/whl/torch_stable.html ``` 如果你选择源码编译,你可以从PyTorch官方网站下载源码并按照官方文档中的指示进行编译和安装。 3. 安装完成后,你可以在Anaconda Prompt或命令提示符中输入以下命令来检测是否成功安装了PyTorch: ``` python -c "import torch; print(torch.__version__)" ``` 如果成功安装,你将看到PyTorch版本号。 请注意,以上步骤仅适用于安装CPU版本PyTorch。如果你想安装GPU版本PyTorch,你需要确保你的电脑上有兼容的NVIDIA显卡,并按照官方文档中的指示进行安装和配置。 #### 引用[.reference_title] - *1* *2* [PyTorchCPU版本的安装步骤](https://blog.youkuaiyun.com/haoxuejiebetter/article/details/129698765)[target="_blank" data-report-click={"spm":"1018.2226.3001.9630","extra":{"utm_source":"vip_chatgpt_common_search_pc_result","utm_medium":"distribute.pc_search_result.none-task-cask-2~all~insert_cask~default-1-null.142^v91^insertT0,239^v3^insert_chatgpt"}} ] [.reference_item] - *3* [【pytorch-cpu版本安装(以win11系统在anaconda虚拟环境为例】](https://blog.youkuaiyun.com/m0_58215969/article/details/127131614)[target="_blank" data-report-click={"spm":"1018.2226.3001.9630","extra":{"utm_source":"vip_chatgpt_common_search_pc_result","utm_medium":"distribute.pc_search_result.none-task-cask-2~all~insert_cask~default-1-null.142^v91^insertT0,239^v3^insert_chatgpt"}} ] [.reference_item] [ .reference_list ]
评论 2
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值