【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

点赞(0) 打赏

评论列表 共有 0 条评论

暂无评论

微信公众账号

微信扫一扫加关注

发表
评论
返回
顶部