Issues Installing PyTorch 2.7.0 and torchvision 0.17.0 with CUDA 12.9 on Windows 11 - Missing Wheel Files and CUDA Not Enabled

Hello everyone,

I'm trying to install PyTorch 2.7.0 and torchvision 0.17.0 with CUDA support on Windows 11, but I'm facing several issues. I need to align with the versions used by my team’s engineers and ultimately run a project called FramePack. I’d really appreciate your help!

My Environment

Operating System: Windows 11 Pro x64

Python Version: 3.10.6

GPU: NVIDIA GeForce RTX 5070 Ti

NVIDIA Driver Version: 576.02

CUDA Version: 12.9 (as shown by nvidia-smi)

Virtual Environment: C:\Users\JXin\Documents\framepack_cu126_torch26\venv

My Goal

Install PyTorch 2.7.0 and torchvision 0.17.0 with CUDA 12.8 support (compatible with CUDA 12.9).

Ensure PyTorch enables CUDA (currently getting "Torch not compiled with CUDA enabled").

Install Triton 3.3.0 and SageAttention 2.1.1 (need Windows-compatible wheel files).

Resolve the NumPy initialization warning: UserWarning: Failed to initialize NumPy: _ARRAY_API not found.

What I've Done

Environment Cleanup:

Removed all WSL2 and Ubuntu residuals (used wsl --unregister and deleted ext4.vhdx).

Cleaned the system-level Python environment (C:\Users\JXin\AppData\Local\Programs\Python\Python310), removing all torch and nvidia-* related packages.

Created a virtual environment: python -m venv venv.

Attempted PyTorch Installation:

Used the command:

pip install torch==2.7.0 torchaudio==2.7.0 --index-url https://download.pytorch.org/whl/cu128 --no-cache-dir
pip install torchvision==0.17.0 --no-cache-dir

Result: Installed CPU versions (2.7.0+cpu, 0.17.0+cpu), not the expected 2.7.0+cu128.

Manual Wheel File Download:

Downloaded:

torch-2.7.0+cu128-cp310-cp310-win_amd64.whl

torchaudio-2.7.0+cu128-cp310-cp310-win_amd64.whl

Installed:

pip install C:\Users\JXin\Downloads\torch-2.7.0+cu128-cp310-cp310-win_amd64.whl
pip install C:\Users\JXin\Downloads\torchaudio-2.7.0+cu128-cp310-cp310-win_amd64.whl

Verified:

python -c "import torch; print(torch.__version__); print(torch.cuda.is_available())"

Output:

2.7.0+cu128
True

Confirmed PyTorch now supports CUDA.

Issue: Could Not Find torchvision Wheel File:

Could not find torchvision-0.17.0+cu128-cp310-cp310-win_amd64.whl at https://download.pytorch.org/whl/cu128.

Tried using pip to download automatically:

pip install torchvision==0.17.0 --index-url https://download.pytorch.org/whl/cu128 --no-cache-dir

Result: Still installed CPU version (0.17.0+cpu).

NumPy Warning:

During verification, I get:

UserWarning: Failed to initialize NumPy: _ARRAY_API not found

Current NumPy version might not be fully compatible with PyTorch 2.7.0.

My Questions

Cannot Find torchvision-0.17.0+cu128-cp310-cp310-win_amd64.whl:

Does anyone know where I can download this wheel file, or if there’s an alternative version (e.g., 0.18.0 or 0.16.0) that’s compatible with PyTorch 2.7.0?

If using pip to download automatically, do I need to adjust the --index-url?

NumPy Warning:

How can I resolve UserWarning: Failed to initialize NumPy: _ARRAY_API not found? Should I upgrade NumPy to a specific version (e.g., 2.0.0)?

SageAttention Wheel File:

I need sageattention-2.1.1+cu128torch2.7.0+cp310+cp310-win_amd64.whl (Windows version).

I only found sageattention-2.1.1+cu126torch2.6.0-cp310-cp310-win_amd64.whl (for PyTorch 2.6.0 and CUDA 12.6). Is this compatible? If not, where can I find the correct version?

What I've Tried

Verified CUDA environment:

nvidia-smi shows CUDA 12.9, driver is working.

Environment variable CUDA_PATH is set to C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v12.9.

Tried alternative torchvision versions (e.g., 0.18.0 and 0.16.0), but I'm concerned about compatibility with PyTorch 2.7.0.

Upgraded NumPy:

pip install --force-reinstall numpy==2.0.0

But the warning persists.

Request for Help

How can I resolve the issue of not finding torchvision-0.17.0+cu128-cp310-cp310-win_amd64.whl?

Is there another way to install PyTorch 2.7.0 and torchvision 0.17.0 with CUDA 12.8/12.9 on Windows 11?

How can I fix the NumPy warning?

Where can I find the correct SageAttention 2.1.1 wheel file?

You can also reach me directly at kaworukevin@gmail.com if you have any suggestions or files to share.

Thanks for your help!

Start by reproducing the listed pip commands in a clean Windows 11 Python 3.10 virtual environment, checking the PyTorch cu128 index and the requested torch, torchvision, torchaudio, NumPy, Triton, and SageAttention versions. Done means documenting a confirmed compatible package set and verification output showing the intended versions and CUDA availability; the issue names no repository file or test to inspect.

Written by the indexing model from the issue text.