RTX5060显卡安装cuda版本PyTorch踩坑记录

本地环境

显卡：RTX5060：

python版本：3.10.18

CUDA版本：12.9

nvidia-smi

参考RagFlow

我在RagFlow上装了torch，因此我猜只要装此版本的应该就OK了

pip install torch2.6.0 torchvision0.21.0 torchaudio==2.6.0 --index-url https://download.pytorch.org/whl/cu124

安装成功？

print(torch.cuda.is_available())

执行下面代码报错：执行失败

RuntimeError: CUDA error: no kernel image is available for execution on the device

>>> import torch
>>> torch.tensor([1.0, 2.0]).cuda()

参考PyTorch安装

如下图所示，找到自己对应的CUDA版本（一定要版本一致）直接安装即可：

参考链接：https://pytorch.org/get-started/locally/

pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu129

安装成功