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

username: kevin_Kaworu
cooked: <p>Hello everyone,</p>
<p>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!</p>
<h3><a name="p-469822-my-environment-1" class="anchor" href="#p-469822-my-environment-1"></a>My Environment</h3>
<ul>
<li><strong>Operating System</strong>: Windows 11 Pro x64</li>
<li><strong>Python Version</strong>: 3.10.6</li>
<li><strong>GPU</strong>: NVIDIA GeForce RTX 5070 Ti</li>
<li><strong>NVIDIA Driver Version</strong>: 576.02</li>
<li><strong>CUDA Version</strong>: 12.9 (as shown by nvidia-smi)</li>
<li><strong>Virtual Environment</strong>: C:\Users\JXin\Documents\framepack_cu126_torch26\venv</li>
</ul>
<h3><a name="p-469822-my-goal-2" class="anchor" href="#p-469822-my-goal-2"></a>My Goal</h3>
<ul>
<li>Install PyTorch 2.7.0 and torchvision 0.17.0 with CUDA 12.8 support (compatible with CUDA 12.9).</li>
<li>Ensure PyTorch enables CUDA (currently getting “Torch not compiled with CUDA enabled”).</li>
<li>Install Triton 3.3.0 and SageAttention 2.1.1 (need Windows-compatible wheel files).</li>
<li>Resolve the NumPy initialization warning: UserWarning: Failed to initialize NumPy: _ARRAY_API not found.</li>
</ul>
<h3><a name="p-469822-what-ive-done-3" class="anchor" href="#p-469822-what-ive-done-3"></a>What I’ve Done</h3>
<ol>
<li><strong>Environment Cleanup</strong>:</li>
</ol>
<ul>
<li>Removed all WSL2 and Ubuntu residuals (used wsl --unregister and deleted ext4.vhdx).</li>
<li>Cleaned the system-level Python environment (C:\Users\JXin\AppData\Local\Programs\Python\Python310), removing all torch and nvidia-* related packages.</li>
<li>Created a virtual environment: python -m venv venv.</li>
</ul>
<ol start="2">
<li><strong>Attempted PyTorch Installation</strong>:</li>
</ol>
<ul>
<li>Used the command:</li>
</ul>
<pre><code class="lang-auto">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
</code></pre>
<ul>
<li>Result: Installed CPU versions (2.7.0+cpu, 0.17.0+cpu), not the expected 2.7.0+cu128.</li>
</ul>
<ol start="3">
<li><strong>Manual Wheel File Download</strong>:</li>
</ol>
<ul>
<li>Downloaded:
<ul>
<li>torch-2.7.0+cu128-cp310-cp310-win_amd64.whl</li>
<li>torchaudio-2.7.0+cu128-cp310-cp310-win_amd64.whl</li>
</ul>
</li>
<li>Installed:</li>
</ul>
<pre><code class="lang-auto">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
</code></pre>
<ul>
<li>Verified:</li>
</ul>
<pre><code class="lang-auto">python -c "import torch; print(torch.__version__); print(torch.cuda.is_available())"
</code></pre>
<p>Output:</p>
<pre><code class="lang-auto">2.7.0+cu128
True
</code></pre>
<pre><code>* Confirmed PyTorch now supports CUDA.
</code></pre>
<ol start="4">
<li><strong>Issue: Could Not Find</strong> torchvision <strong>Wheel File</strong>:</li>
</ol>
<ul>
<li>Could not find torchvision-0.17.0+cu128-cp310-cp310-win_amd64.whl at <a href="https://download.pytorch.org/whl/cu128" rel="noopener nofollow ugc">https://download.pytorch.org/whl/cu128</a>.</li>
<li>Tried using pip to download automatically:</li>
</ul>
<pre><code class="lang-auto">pip install torchvision==0.17.0 --index-url https://download.pytorch.org/whl/cu128 --no-cache-dir
</code></pre>
<p>Result: Still installed CPU version (0.17.0+cpu).<br>
5. <strong>NumPy Warning</strong>:</p>
<ul>
<li>During verification, I get:</li>
</ul>
<pre><code class="lang-auto">UserWarning: Failed to initialize NumPy: _ARRAY_API not found
</code></pre>
<ul>
<li>Current NumPy version might not be fully compatible with PyTorch 2.7.0.</li>
</ul>
<h3><a name="p-469822-my-questions-4" class="anchor" href="#p-469822-my-questions-4"></a>My Questions</h3>
<ol>
<li><strong>Cannot Find</strong> torchvision-0.17.0+cu128-cp310-cp310-win_amd64.whl:</li>
</ol>
<ul>
<li>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?</li>
<li>If using pip to download automatically, do I need to adjust the --index-url?</li>
</ul>
<ol start="2">
<li><strong>NumPy Warning</strong>:</li>
</ol>
<ul>
<li>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)?</li>
</ul>
<ol start="3">
<li><strong>SageAttention Wheel File</strong>:</li>
</ol>
<ul>
<li>I need sageattention-2.1.1+cu128torch2.7.0+cp310+cp310-win_amd64.whl (Windows version).</li>
<li>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?</li>
</ul>
<h3><a name="p-469822-what-ive-tried-5" class="anchor" href="#p-469822-what-ive-tried-5"></a>What I’ve Tried</h3>
<ul>
<li>Verified CUDA environment:
<ul>
<li>nvidia-smi shows CUDA 12.9, driver is working.</li>
<li>Environment variable CUDA_PATH is set to C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v12.9.</li>
</ul>
</li>
<li>Tried alternative torchvision versions (e.g., 0.18.0 and 0.16.0), but I’m concerned about compatibility with PyTorch 2.7.0.</li>
<li>Upgraded NumPy:</li>
</ul>
<pre><code class="lang-auto">pip install --force-reinstall numpy==2.0.0
</code></pre>
<p>But the warning persists.</p>
<h3><a name="p-469822-request-for-help-6" class="anchor" href="#p-469822-request-for-help-6"></a>Request for Help</h3>
<ul>
<li>How can I resolve the issue of not finding torchvision-0.17.0+cu128-cp310-cp310-win_amd64.whl?</li>
<li>Is there another way to install PyTorch 2.7.0 and torchvision 0.17.0 with CUDA 12.8/12.9 on Windows 11?</li>
<li>How can I fix the NumPy warning?</li>
<li>Where can I find the correct SageAttention 2.1.1 wheel file?</li>
</ul>
<p>Thanks for your help!</p>

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

username: itzolkmar
cooked: <p>I’ve got a new system with a Geforce 5090 and no AI tool worked. When I found this helpful hints of Kevin and ptrblck I tried it with Vlad Mandic’s SDNext and it worked very fine with Flux.1 Dev. SageAttention is not necessary there.</p>
<p>I use Python3-11 and therefore loaded the following packages from <a href="https://download.pytorch.org/whl:" rel="noopener nofollow ugc">https://download.pytorch.org/whl:</a></p>
<p>torch-2.7.0+cu128-cp311-cp311-win_amd64.whl<br>
torchaudio-2.7.0+cu128-cp311-cp311-win_amd64.whl<br>
torchvision-0.22.0+cu128-cp311-cp311-win_amd64.whl</p>
<p>Installation worked easily:<br>
.\pip install ‘\torch 2.7\torch-2.7.0+cu128-cp311-cp311-win_amd64.whl’<br>
.\pip install ‘\torch 2.7\torchaudio-2.7.0+cu128-cp311-cp311-win_amd64.whl’<br>
.\pip install ‘\torch 2.7\torchvision-0.22.0+cu128-cp311-cp311-win_amd64.whl’</p>
<p>It seems torchaudio has already been installed with torch, cause it was yet found.</p>
<p>SDNext produced a nice apple picture for a test with Flux.1 Dev.<br>
Thanks guys! You made my day!  <img src="https://discuss.pytorch.org/images/emoji/apple/grinning_face.png?v=14" title=":grinning_face:" class="emoji" alt=":grinning_face:" loading="lazy" width="20" height="20"></p>