RT5080安装cuda并且pytorch使用GPU时问题记录

1.安装cuda 12.8版本(只能安装12.8版本，低于这个版本不兼容，会报错)---------------------以下所有操作都需要梯子和大量的流量大概需要至少7G左右---甚至10
https://developer.nvidia.com/cuda-downloads?target_os=Windows&target_arch=x86_64&target_version=11

 

 2.安装cuDNN啥的（不确定有没有用，但是还是装上了）

https://developer.nvidia.com/rdp/cudnn-archive

选择的适用12版本的

3.最关键的安装Pytorch，

pip install --force-reinstall torch==2.7.0.dev20250307+cu128 torchvision==0.22.0.dev20250308+cu128 torchaudio==2.6.0.dev20250308+cu128 --index-url https://download.pytorch.org/whl/nightly/cu128 --extra-index-url https://download.pytorch.org/whl/nightly/cu128