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,则表示安装成功了!
进行到这里,表示你已经全部安装成功了,祝贺!!!

Comments NOTHING