显卡算力与Pytorch版本不匹配

跟着李沐老师跑LeNet的代码时报错
NVIDIA GeForce RTX 3080 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 3080 GPU with PyTorch, please check the instructions at https://pytorch.org/get-started/locally/
本人配置情况：
CUDA11.6
Python3.8
Torch1.12.1+cu102

到Pytorch官网查看对应版本。

使用上图pip安装之后，仍然报错。
再次查看版本。

import torch
print(torch.__version__)

显示还是1.12.1+cu102没有变化。
最后创建了一个虚拟环境，在虚拟环境里使用conda安装指定版本的Pytorch，和CUDA11.6匹配，问题解决。