安装cuda版本的torch

确保本地显卡是N卡，已经安装了最新驱动。
下载并安装： cuda_12.6.3_561.17_windows.exe
下载并解压到上面cuda的目录： cudnn-windows-x86_64-9.3.0.75_cuda12-archive.zip
下载并安装： Miniconda3-latest-Windows-x86_64.exe（最好选中设置环境变量，虽然它不建议），修改源为清华，具体忘了哪个命令。
conda create -n torch
conda activate torch
进入https://pytorch.org/get-started/previous-versions/，找到Linux and Windows，看到下面的命令

# CUDA 12.4
pip install torch==2.5.0 torchvision==0.20.0 torchaudio==2.5.0 --index-url https://download.pytorch.org/whl/cu124

所以参照上面，在conda中输入

conda install pytorch==2.5.0 torchvision==0.20.0 torchaudio==2.5.0 pytorch-cuda=12.4 -c pytorch -c nvidia

完成之后，在python中输入下面代码，显示版本和是否支持CUDA

import torch
print(torch.__version__)    #显示版本
print(torch.cuda.is_available())  #

另外的命令
conda env list