2.3.1 Install requirements (1) Install pytorch 2.3.1 (For more details, please refer to https://pytorch.org/) For linux: # CUDA 11.8 $ conda install pytorch==2.3.1 torchvision==0.18.1 torchaudio==2.3.1 pytorch-