Install pytorch according to your system configuration. The code is tested on torch v2.0.1, torchvision 0.15.2.

安装PyTorch需要根据你的系统配置来选择合适的版本和依赖。对于测试过的torch版本v2.0.1和torchvision 0.15.2，你可以参考以下步骤来进行安装：

**对于Windows用户：**
1. 访问PyTorch官网 <https://pytorch.org/get-started/locally/> 并下载适用于Windows的.whl文件。
2. 使用`pip install`命令安装，例如：

   pip install torch==2.0.1+cu92 torchvision==0.15.2+cu92

确保替换成你的CUDA版本（这里假设是cu92）。

**对于macOS用户：**
1. 安装anaconda（如果还未安装）：<https://www.anaconda.com/products/distribution>
2. 打开终端并运行：

   conda create -n pytorch python=3.6 torchvision cudatoolkit=9.2 -c pytorch
   conda activate pytorch
   pip install torch torchvision==0.15.2

**对于Linux用户（Ubuntu/Debian）：**
1. 更新软件包列表：

   sudo apt-get update

2. 安装必要的库：

   sudo apt-get install python3-dev cmake libncurses5-dev libopenblas-dev libfreetype6-dev libpng-dev libjpeg-dev libhdf5-dev protobuf-compiler liblapack-dev gfortran

3. 获取并安装PyTorch：

   git clone https://github.com/pytorch/torch.git
   cd torch
   git checkout v2.0.1
   export CUDA_HOME=<your_CUDA_home_directory>
   python setup.py install

4. 同样安装torchvision：

   git clone https://github.com/pytorch/vision.git
   cd vision
   git checkout v0.15.2
   pip install -e .

确保替换上述命令中的`<your_CUDA_home_directory>`为你实际的CUDA安装目录。

安装完成后，你可以通过`import torch`和`import torchvision`来验证是否成功导入这两个库。