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

plain 复制代码
>>> 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

安装成功