0
点赞
收藏
分享

微信扫一扫

【PyTorch】安装pytorch方法总结

晴儿成长记 2024-08-05 阅读 19

【PyTorch】安装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

在【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 条评论