前期准备：安装anaconda、python、vs等

检查cuda和cudnn是否安装成功

nvcc -V

去cuda的安装文件

C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.4\extras\demo_suite

在终端分别运行两个文件，如果出现以下输出表示cudnn安装成功

在anaconda 终端安装pytorch

conda create -n pytorch python=3.8
conda activate pytorch
conda install pytorch torchvision torchaudio cudatoolkit=11.3 -c pytorch

在虚拟环境里使用jupyter notebook

conda install ipykernel

python -m ipykernel install --name pytorch

然后就可以在jupyter notebook里写pytorch了，不过还是推荐用vscode写:)