安装使用pytorch
2023-11-21 本文已影响0人
wo_monic
创建conda环境
conda create -n pytorch
conda activate pytorch
注意下面的安装的都是linux的cpu版本的pytorch,更多版本和命令参考官网
使用conda安装pytorch
conda install pytorch torchvision torchaudio cpuonly -c pytorch
或者使用pip安装pytorch
pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cpu
把环境添加到jupyter的环境里
conda install ipykernel
python -m ipykernel install --user --name pytorch --display-name "pytorch"
检测是否安装成功
import torch
x = torch.rand(3, 2)
print(x)
输出数字的矩阵，表示安装成功，报错说明安装没有成功。
使用conda安装报错，找不到pytorch channel,使用下面的命令添加中科大的conda源，再安装即可。
conda config --add channels https://mirrors.ustc.edu.cn/anaconda/cloud/pytorch/
conda install pytorch torchvision torchaudio cpuonly pytorch
pytorch入门
1. 张量tensor
1个数就是标量，0维张量
1列数就是矢量，1维张量
1个数据表，矩阵，2维张量
1堆数据表，矩阵数组，3维张量
低纬度的张量
用正方体表示3维张量
image.png
一列正方体，就是4维张量
一片正方体，就是5维张量
image.png
