5060显卡、cuda、pytorch兼容问题

Question

UserWarning: 
NVIDIA GeForce RTX 5060 Laptop GPU with CUDA capability sm_120 is not compatible with the current PyTorch installation.
The current PyTorch install supports CUDA capabilities sm_50 sm_60 sm_61 sm_70 sm_75 sm_80 sm_86 sm_37 sm_90 compute_37.
If you want to use the NVIDIA GeForce RTX 5060 Laptop GPU GPU with PyTorch, please check the instructions at https://pytorch.org/get-started/locally/

  warnings.warn(

虚心请教，目前是初学者 请问这种应该怎么解决啊 我目前下载的是cuda 11.8版本，pytorch 2.3.1版本

Answer 1

阿里嘎多学长整理AIGC生成，因移动端显示问题导致当前答案未能完全显示，请使用PC端查看更加详细的解答过程

问题解答

你遇到的问题是 PyTorch 在使用 NVIDIA GeForce RTX 5060 Laptop GPU 时出现了不兼容警告。警告信息中提到 sm_120 是不兼容的 CUDA 能力。

解决方案：

1. 检查 PyTorch 的版本是否支持你的 GPU。可以使用以下命令检查：

import torch
print(torch.cuda.get_device_name(0))

如果你的 GPU 不在支持列表中，可以升级 PyTorch 到最新版本。

1. 如果你的 GPU 在支持列表中，但仍然出现不兼容警告，可以尝试使用以下命令来强制使用你的 GPU：

import torch
torch.cuda.set_device(0)

1. 如果上述方法不行，可以尝试使用 torch.backends.cudnn.enabled = False 来禁用 cuDNN，可能会解决问题。

2. 如果仍然出现问题，可以尝试使用 torch.cuda.device_count() 和 torch.cuda.device() 函数来检查 GPU 的数量和类型。

注意：如果你的 GPU 不在 PyTorch 支持列表中，可能需要等待 PyTorch 的更新或使用其他深度学习框架。