一、使用pip自动安装:
1. 使用 nvidia-smi 命令查看服务器对应的cuda版本及驱动版本,如下图所示:
2.根据对应cuda的版本,去pytorch官网Previous PyTorch Versions | PyTorch
,按照指定命令进行安装【测试发现使用pip命令能够成功安装】:
pip install torch==2.2.0 torchvision==0.17.0 torchaudio==2.2.0 --index-url https://download.pytorch.org/whl/cu121
二、手动下载安装:
如果有服务器无法联网,只能手动下载的情况,可以先到对应的镜像源(如清华镜像源)下载对应的torch+cuda+cudnn,然后放到本地,使用conda或者pip命令进行安装