1、判断要安装CPU还是GPU版本
右键开始菜单,在弹出选项中选择任务管理器→选择性能。看GPU是否带NVIDIA的英伟达显卡,没有的话就只能安装CPU版本。
两个显卡但都不是英伟达版本。
2、安装Pytorch
此时你已经安装了Anaconda
1. 创建虚拟环境,激活环境
conda create -n mypytorch python==3.9
conda activate mypytorch
2. 在虚拟环境中安装Pytorch(CPU版本)
使用conda命令。登录官网PyTorch ,选择系在的版本、系统等,选择Conda命令,最后会出现对应语句,复制可用。
conda install pytorch torchvision torchaudio cpuonly -c pytorch
添加镜像
conda install pytorch torchvision torchaudio cpuonly -c 镜像地址
镜像地址,下载快一点
阿里巴巴镜像
http://mirrors.aliyun.com/anaconda/cloud/pytorch/win-64/
清华镜像
https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/win-64/南京大学镜像
https://mirror.nju.edu.cn/pub/anaconda/cloud/pytorch/win-64/北京外国语大学镜像
https://mirrors.bfsu.edu.cn/anaconda/cloud/pytorch/win-64/
下载过程中,在做yes no判断前,可以看要下载的包,找到pytorch,可以分别看到pytorch的版本,对应的python版本、CPU版的,镜像。
成功的展示图。
最后,在Anaconda Prompt输入 conda list 可以看到所有的库
3、验证pytorch是否安装成功
激活对应的虚拟环境环境
conda activate 环境名字
验证
1.输入conda list,看有没有pytorch或者torch
2. 输入python输入import torch
输入torch.cuda.is_available()
最后的命令是验证我们的电脑是否安装了CUDA,我们是CPU版本,肯定没有。如果显示False,就说明我们这个PyTorch安装成功了