Pytorch 安装(CPU)
注意:由于本人的电脑是集体显卡,只能装CPU版本的,无法安装GPU版本的。
一、安装Anaconda
大家可以去anaconda官网进行下载安装:https://www.anaconda.com/products/distribution
这里默认已经安装了Anaconda。
验证anaconda是否安装成功:查看版本号,如下图,则安装成功:
第二步:安装 Pytorch
Pytorch 官网:https://pytorch.org/
在Pytorch官网,找到下图:
若电脑为集体显卡或者显卡为AMD,直接选择CPU版本。选择之后,会有相应的安装命令:
二、安装Pytorch
1、创建虚拟环境:
打开 Anaconda Prompt 输入相应的命令:
输入命令:
conda create --name pytorch python=3.10
其中:pytorch:是虚拟环境的名字; 3.10 是指python的版本号
(都可以根据自己的需求更改环境名字、python版本号)
如下图所示:
输入 y,则继续开始:
该虚拟环境创建成功。
2、激活该环境:
输入命令:conda activate pytorch
3、 在新创建的环境下安装 Pytorch:
该命令为cpu版本的命令,其他版本可参考官网:
pip install torch torchvision torchaudio
4、验证 Pytorch 是否安装成功;
查看 pytorch 版本:若出现则说明安装成功。
print (torch.__version__)
注意:是两个下划线。
再次验证是否成功安装:
import torch
x = torch.rand(5,3)
print(x)
两次验证,均可说明安装成功。
三、删除虚拟环境
在当前环境中,不能删除当前的虚拟环境,需切换到其他环境进行删除。
则,提示不能删除该环境。
切换到其他环境下:activate base
在该环境下输入相应命令:
conda remove -n zmy --all
其中:zmy:为要删除环境的名称
zmy环境以删除成功。
四、查看所有环境
输入命令:
conda info -e
其中:* 代表当前所在的环境。
加油! 每天收获一点点!