硬件环境 WSL Ubuntu 系统，NVIDIA 显卡。
一、准备工作
首先，在 NVIDIA CUDA GPU 查看您的 GPU 是否支持 CUDA。
保证当前 Linux 版本支持 CUDA。在命令行中输入：
uname -m && cat /etc/*release
输出如下，不一定完全一样，类似即可：
检查是否安装了 gcc。在命令行中输入 gcc --version，应当看到类似的输出：
二、PyTorch 和 CUDA 安装
使用官网版本 2.6.0，官网链接：Previous PyTorch Versions。
根据其指令安装对应版本（以 CUDA 12.4 为例）：
pip install torch==2.6.0 torchvision==0.21.0 torchaudio==2.6.0 --index-url https://download.pytorch.org/whl/cu124
三、校验 Torch 和 CUDA 版本
安装完查看自己的 torch 版本：
conda list
cuda 版本则需要进入 python：
import torch
print(torch.version.cuda)
