【PyTorch】安装pytorch方法总结
【PyTorch】安装pytorch方法总结
一、官网指令安装
这种方法可能响应时间较长,此时可以使用其他方法
二、远程下载包至本地,本地安装
参考:pip安装python包(pytorch)时遇到超时现象的通用解决方案
在【https://download.pytorch.org/whl/torch/】上找到pytorch版本、python版本、cuda版本、操作系统合适的包(下载到的包后缀为whl),然后使用下方命令安装(linux可用):
wget -c https://download.pytorch.org/whl/torch-2.1.2+cu121-cp310-cp310-linux_x86_64.whl
pip install torch-2.1.2+cu121-cp310-cp310-linux_x86_64.whl
此方法响应也慢的话,可以使用其他方法
三、使用清华镜像源下载pytorch
参考:使用清华镜像源下载pytorch
在【https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/】上找到合适的包(下载到的包后缀为tar.bz2),然后使用下方命令安装(linux可用):
wget https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/linux-64/pytorch-2.1.2-py3.10_cuda12.1_cudnn8.9.2_0.tar.bz2
conda install --use-local pytorch-2.1.2-py3.10_cuda12.1_cudnn8.9.2_0.tar.bz2