树莓派学习(三):安装pytorch并验证

树莓派学习(三):安装pytorch并验证标题

  • 步骤一:创建一个虚拟环境
  • 步骤二:安装pythorch
  • 步骤三:测试
  • 彩蛋

步骤一:创建一个虚拟环境

conda create --name pytorchcpu python==3.9.10

树莓派学习(三):安装pytorch并验证_第1张图片
树莓派学习(三):安装pytorch并验证_第2张图片
如果需要删除虚拟环境,输入以下指令
conda remove -n (虚拟环境名称) --all

步骤二:安装pythorch

  • (1)激活base虚拟环境
> conda activate base #先输入此命令,不然无法激活
> conda info --envs

树莓派学习(三):安装pytorch并验证_第3张图片
在激活后,应该出现(base)字样,然后,查看存在的虚拟环境,发现多了一个pytorchcpu

conda activate pytorchcpu

在这里插入图片描述
当前环境变成了pytorchcpu

  • (2)安装pytorch

在官网安装pytorch
树莓派学习(三):安装pytorch并验证_第4张图片
按环境配置选信息,然后复制命令,到激活的环境中下载

如果下载失败,可以使用国内镜像源

pip3 install torch torchvision torchaudio -i https://pypi.tuna.tsinghua.edu.cn/simple

步骤三:测试

1.我已经将网络模型在windows下训练好,并且将测试文件和所需模型等传到树莓派
2.该模型负责识别FashionMNIST数据集,接下来测试效果

树莓派学习(三):安装pytorch并验证_第5张图片
在这里插入图片描述

预测效果也是正确的

彩蛋

我们虽然创建了虚拟环境,但是代码不知道啊,因此,我们需要在.py文件最开头添加你工具包的路径

import sys
sys.path.append("/home/lh/.conda/envs/pytorchcpu/lib/python3.9/site-packages")

打个小广告
有想学习深度学习pytorch框架的朋友,可以参考本人相关仓库

你可能感兴趣的:(树莓派学习,pytorch,学习,python,linux)