官网地址：https://pytorch.org/

一、访问官网，获取安装命令

访问地址：https://pytorch.org/get-started/locally

选择要安装的环境配置，复制命令

我的命令为：

conda install pytorch::pytorch torchvision torchaudio -c pytorch

二、在终端运行命令

如果没有安装 conda，请先安装，参考

1、新建虚拟环境

conda create -n pytorch2.2.1 python=3.7

2、激活虚拟环境

conda activate pytorch13

3、安装pytorch

我的Mac下命令是

conda install pytorch::pytorch torchvision torchaudio -c pytorch

我的Windows命令是（使用 nvidia-smi 命令查看CUDA版本，我的为11.6）

pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118

4、进入python命令窗口

python

5、查看pytorch版本号

import torch
print(torch.__version__)

6、计算一个矩阵计算

N, D = 3, 4
x = torch.tensor(torch.rand(N,D))
y = torch.tensor(torch.rand(N,D))
z = torch.tensor(torch.rand(N,D))
a = x*y
b = a+z
c = torch.sum(b)
c