pytorch坑(2),GPU和CUDA库的版本不匹配

pytorch坑(2),GPU和CUDA库的版本不匹配_第1张图片

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

pytorch坑(2),GPU和CUDA库的版本不匹配_第2张图片

 

你可能感兴趣的:(pytorch,深度学习,人工智能)