在命令行输入：nvidia-smi ，查看CUDA版本，这里看到我的CUDA版本为12.0，这里CUDA版本主要是影响安装的torch、torchvision、torchaudio三个组件。刚开始我在PyCharm之前设置的环境中重新安装pytorh，但是一直不行，于是我参考了网上解决方法，重新创建了新的虚拟环境，直接在官网上装Pytorch（GPU）版本，解决了问题。创建完pytorch虚拟环境