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).
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
- 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?
Thanks for your help!