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

  1. 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.
  1. 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.
  1. 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.
  1. Issue: Could Not Find torchvision Wheel File:
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).
5. 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

  1. 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?
  1. 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)?
  1. 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?

Thanks for your help!

PyTorch 2.7.0 released with torchvision=0.22.0 while you are trying to install an old vision build which wasn’t built with CUDA 12.8. You might need to build sageattention from source or check the repository to see if the authors built a compatible release.