代码运行遇到报错:
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是否可用
请问是什么原因导致报错呢?
应该是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
你这也没截出cuda是不是10.2啊,看看是不是10.2