出现问题当前cuda的算力与gpu的算力不匹配的问题。NVIDIA GeForce RTX GPU 的算力为8.6
而当前的Pytorch安装的CUDA最高算力支持7.5
查看显卡算力:https://developer.nvidia.com/zh-cn/cuda-gpus
查看CUDA的算力:https://docs.nvidia.com/cuda/ampere-compatibility-guide/index.html
简而言之:GPU的算力需要高于CUDA的算力(但是二者的算力必须在同版本下,比如都是8.x(Gpu sm_8.a>CUDA sm_8.b)。如果GPU的算力为8.x,CUDA的算力为7.x,即使满足>的条件,但也是不行的)
版权声明:本文为CSDN博主「柠檬也可以不酸」的原创文章,遵循CC 4.0 BY-SA版权协议,转载请附上原文出处链接及本声明。
原文链接:https://blog.csdn.net/chengxy1998/article/details/121924208
暂无评论