du制作网站,国内营销策划咨询公司,800元网站建设,乒乓球网站建设目标Geforce 50xx系显卡最低支持cuda128#xff0c;torch cu128 release版本目前还没有释放#xff0c;所以自己基于2.6.0源码自己编译wheel包。
1. 前置条件
1. 使用visual studio installer 安装visual studio 2022#xff0c;工作负荷选择【使用c的桌面开发】,安装完成后将…Geforce 50xx系显卡最低支持cuda128torch cu128 release版本目前还没有释放所以自己基于2.6.0源码自己编译wheel包。
1. 前置条件
1. 使用visual studio installer 安装visual studio 2022工作负荷选择【使用c的桌面开发】,安装完成后将“VC\Tools\MSVC\版本号\bin\Hostx64\x64”对应的路径加入环境变量 2. 访问https://github.com/pytorch/pytorch/releases/download/v2.6.0/pytorch-v2.6.0.tar.gz下载源码可以替换成kkgithub镜像访问下载后解压
3. 访问https://github.com/conda-forge/miniforge/releases/download/24.11.3-0/Miniforge3-Windows-x86_64.exe安装miniforge
2. 编译
打开 Miniforge Prompt依次执行如下代码
##如果访问报错或速度缓慢搜索并替换conda、pip清华源
conda create torch_build_env python3.10 libuv
conda activate torch_build_env
conda install -y astunparse numpy ninja pyyaml setuptools cmake cffi typing_extensions future six requests dataclassesset MAX_JOBS24
set USE_DISTRIBUTEDOFF
set USE_CUDAON
set PYTORCH_BUILD_VERSION2.6.0a0cu128-cp31016-win_x64
set PYTORCH_BUILD_NUMBER0 cd C:\pytorch-v2.6.0
##执行前先确保删除build目录
python setup.py bdist_wheel
编译完成后wheel包在dist目录下使用pip安装
pip install dist\torch-2.6.0a0cu128.cp31016.win.x64-cp310-cp310-win_amd64.whl
