Pytorch安装指南

犀利的毛毛虫 发布于 2025-02-02 232 次阅读


pytorch下载安装

进入pytorch官网:https://pytorch.org/

点击Get Started

复制命令到Anaconda Promote中,需要注意安装的pytorch版本应该小于安装的cuda版本(我的cuda版本为12.8,因此我安装12.6的pytorch):

查看cuda版本的命令为 nvcc --version

要注意显卡的驱动版本应该大于500:

总之,复制上面图片的地址后下载安装即可。此方法是在官网下载,下载过程很慢,下面讲解切换镜像源的下载方法。目前最快的是阿里云的cu121镜像:

这里我们下载cu118版本的。在conda中输入命令:

pip install torch==2.3.1 torchvision==0.18.1 torchaudio==2.3.1 -f https://mirrors.aliyun.com/pytorch-wheels/cu118

下载速度很快。

验证PyTorch是否安装成功

激活你安装的虚拟环境,例如 conda activate pytorch

输入 pip list ,寻找是否有pytorch或者torch,有就表示安装成功。

进一步验证
输入

python
import torch
torch.cuda.is_available()

如果显示True,则表示安装成功了!

进行到这里,表示你已经全部安装成功了,祝贺!!!