win11编译pytorchaudio cuda128版本流程

System_sleep2025-03-04 9:18

1. 前置条件

本篇续接自 win11编译pytorch cuda128版本流程，阅读前请先参考上一篇配置环境。

访问https://kkgithub.com/pytorch/audio/archive/refs/tags/v2.6.0.tar.gz下载源码，下载后解压；

2. 编译

在visual studio 2022安装目录下查找如下文件路径：

VC\Auxiliary\Build\vcvarsall.bat

打开 Miniforge Prompt，依次执行如下代码：

bash 复制代码

conda activate torch_build_env

##替换成自己的路径
call "C:\Program Files (x86)\Microsoft Visual Studio\2019\Community\VC\Auxiliary\Build\vcvarsall.bat" x64

cd c:\audio-2.6.0
##执行前先确认删除build目录
python setup.py bdist_wheel

如果遇到如下ffmpeg下载错误，一般是s3.amazonaws.com无法访问造成：

bash 复制代码

 error: downloading 'https://pytorch.s3.amazonaws.com/torchaudio/ffmpeg/2023-08-14/windows/4.4.4.tar.gz' failed
          status_code: 35
          status_string: "SSL connect error"

FAILED: f4-populate-prefix/src/f4-populate-stamp/f4-populate-download C:/audio-2.6.0/build/temp.win-amd64-cpython-310/Release/_deps/f4-subbuild/f4-populate-prefix/src/f4-populate-stamp/f4-populate-download

可以参考报错日志，手动下载ffmpeg源码包放到指定src目录下。

或者设置代理，让cmake可以访问到网址：

bash 复制代码

set HTTP_PROXY=socket5|http://yourproxy.com:port
set HTTPS_PROXY=socket5|http|https://yourproxy.com:port

编译完成后，wheel包在dist目录下，使用pip安装

bash 复制代码

pip install dist\torchaudio-2.6.0a0-cp310-cp310-win_amd64.whl