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,依次执行如下代码:
conda activate torch_build_env##替换成自己的路径
call "C:\Program Files (x86)\Microsoft Visual Studio\2019\Community\VC\Auxiliary\Build\vcvarsall.bat" x64cd c:\audio-2.6.0
##执行前先确认删除build目录
python setup.py bdist_wheel
如果遇到如下ffmpeg下载错误,一般是s3.amazonaws.com无法访问造成:
error: downloading 'https://pytorch.s3.amazonaws.com/torchaudio/ffmpeg/2023-08-14/windows/4.4.4.tar.gz' failedstatus_code: 35status_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可以访问到网址:
set HTTP_PROXY=socket5|http://yourproxy.com:port
set HTTPS_PROXY=socket5|http|https://yourproxy.com:port
编译完成后,wheel包在dist目录下,使用pip安装
pip install dist\torchaudio-2.6.0a0-cp310-cp310-win_amd64.whl