NVIDIA GeForce RTX ***CUDA capability sm_86 is not compatible with the current PyTorch installation.

错误展示:
NVIDIA 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_70.
If you want to use the NVIDIA GeForce RTX 3070 GPU with PyTorch.
错误描述:
这个错误就是因为你的pytorch(torch)版本太低,不支持高算力的显卡运算。
说的通俗点,你的显卡算力比较高,和低版本的torch不兼容了。
解决办法:
换显卡?换个弱点的显卡?那必不可能啊,hhh,只需要装个高版本的pytorch和torch还有cuda就可以。装啥版本呢?
3070我建议直接最新,cudatoolkit 11.3 torch1.10.0
torch官网链接:https://pytorch.org/get-started/locally/
cudatoolkit安装:

conda install cudatoolkit=11.3
conda install cudnn

如果你还需要装geometric啥的,那就需要将原先的一些torch-scatter之类的包先卸载掉,然后再去官网装
官网链接:https://pytorch-geometric.readthedocs.io/en/latest/notes/installation.html
至此,算力和torch版本不兼容的事情就解决了,但是最新版本的torch后期可能会遇到一些奇怪的现象,如果遇到的话,欢迎再来评论区一起讨论。

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