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

最新推荐文章于 2025-09-11 09:56:32 发布

原创

最新推荐文章于 2025-09-11 09:56:32 发布 · 3.5k 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#pytorch #深度学习 #人工智能

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之后出了问题