Just wanted to paste my working setup in case anyone was having issues or missing anything that keeps causing errors….
Working setup: RTX 5070 Ti (Blackwell, sm_120) + Windows + PyTorch
Hardware
-
GPU: NVIDIA GeForce RTX 5070 Ti
-
Driver: 576.88
-
Reported CUDA (driver): 12.9 (nightly PyTorch cu128 works; drivers are backward compatible)
OS
- Windows 11 (WDDM)
Python
-
Python 3.12.10
-
torch
2.11.0.dev20260110+cu128 -
torchvision
0.25.0.dev20260110+cu128 -
torchaudio
2.11.0.dev20260110+cu128
Verification
-
arch
['sm_70','sm_75','sm_80','sm_86','sm_90','sm_100','sm_120'] -
cuda
True -
gpu
NVIDIA GeForce RTX 5070 Ti -
matmul ok
torch.Size([1024, 1024])
For windows 10 and up use this to get the correct set up, but make sure its all one line if using CMD. Install:
pip install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu128