安装pytorch环境

# CPU 安装;比较简单;指定对应版本即可
pip install torch==2.3.1 torchvision==0.18.1 torchaudio==2.3.1

# 如果需要安装最新的版本,如下操作
pip install torch torchvision torchaudio

# 但是还是比较建议指定一下版本,避免兼容性的问题~

# GPU 安装;需要根据CUDA版本来选择对应的版本;比如我的显卡驱动支持12.1,那安装方法如下
pip install torch==2.3.1+cu121 torchvision==0.18.1+cu121 torchaudio==2.3.1+cu121 -f https://download.pytorch.org/whl/torch_stable.html
# 这里为什么要指定下载源呢~ 因为国内大多数源并没有GPU的版本,会报错找不到对应的版本~