cuda与torch匹配问题

Question

代码运行遇到报错：

UserWarning: 
NVIDIA GeForce RTX 3050 Ti Laptop GPU with CUDA capability sm_86 is not compatible with the current PyTorch installation.
The current PyTorch install supports CUDA capabilities sm_37 sm_50 sm_60 sm_61 sm_70 sm_75 compute_37.

但是我的cuda与torch是相匹配的

cuda版本

检验GPU是否可用

请问是什么原因导致报错呢？

Answer 1

应该是RTX3050Ti是版本比较新，所以cuda需要11.X 可以参考https://blog.csdn.net/zc19981224/article/details/119705267?ops_request_misc=&request_id=&biz_id=102&utm_term=rtx3050ti%E6%94%AF%E6%8C%81%E7%9A%84cuda%E7%89%88%E6%9C%AC&utm_medium=distribute.pc_search_result.none-task-blog-2~all~sobaiduweb~default-3-119705267.nonecase&spm=1018.2226.3001.4187