一，安装cuda

直接去官网下载安装cuda，我安装的是12.1版本的

python -c "import torch; print(torch.__version__)"
python -c "import torchvision; print(torchvision.__version__)"
python -c "import torchaudio; print(torchaudio.__version__)"

二，在pycharm终端里面运行一下代码查看返回结果

python -c "import torch; print(torch.__version__)"
python -c "import torchvision; print(torchvision.__version__)"
python -c "import torchaudio; print(torchaudio.__version__)"

一般第一次返回结果三个不会同时是什么什么+cu，因为我已经配好了所以我的是什么什么＋cu121

三，先把版本不符的卸载一下，运行下面命令

pip uninstall torch torchvision torchaudio

四，安装符合版本的 torch  torchvision  torchaudio（我的cuda版本是12.1，如果cuda版本不一样，需要自己查找相关符合版本的，不能直接复制下面代码）

pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu124

五，安装完成后，在运行一遍第二步的三行代码，查看返回结果，如果返回结果三个都是什么什么+cu，就代表环境配置成功。