pytorch坑（2）,GPU和CUDA库的版本不匹配

最新推荐文章于 2024-09-28 19:24:12 发布

种豆得瓜er

最新推荐文章于 2024-09-28 19:24:12 发布

阅读量3.2k

点赞数

文章标签： pytorch 深度学习人工智能

本文链接：https://blog.csdn.net/qq_41166909/article/details/122531712

版权

E:\Anaconda3\envs\pytorch\lib\site-packages\torch\cuda\__init__.py:143: UserWarning: 
GeForce RTX 3070 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.
If you want to use the GeForce RTX 3070 GPU with PyTorch, please check the instructions at Start Locally | PyTorch

  warnings.warn(incompatible_device_warn.format(device_name, capability, " ".join(arch_list), device_name))

在网上查到的答案是GPU的算力要和CUDA库相匹配。否则GPU无法运行

colab 上，pytorch的1.10.0配的是cu111,我以前在自己的电脑上运行的时候一点问题都没有，我推测是自己升级了pytorch到1.10.1之后出了问题