树莓派学习(三):安装pytorch并验证标题
步骤一:创建一个虚拟环境
conda create --name pytorchcpu python==3.9.10
如果需要删除虚拟环境,输入以下指令
conda remove -n (虚拟环境名称) --all
步骤二:安装pythorch
- (1)激活base虚拟环境
> conda activate base #先输入此命令,不然无法激活
> conda info --envs
在激活后,应该出现(base)字样,然后,查看存在的虚拟环境,发现多了一个pytorchcpu
conda activate pytorchcpu
当前环境变成了pytorchcpu
- (2)安装pytorch
在官网安装pytorch
按环境配置选信息,然后复制命令,到激活的环境中下载
如果下载失败,可以使用国内镜像源
pip3 install torch torchvision torchaudio -i https://pypi.tuna.tsinghua.edu.cn/simple
步骤三:测试
1.我已经将网络模型在windows下训练好,并且将测试文件和所需模型等传到树莓派
2.该模型负责识别FashionMNIST数据集,接下来测试效果
预测效果也是正确的
彩蛋
我们虽然创建了虚拟环境,但是代码不知道啊,因此,我们需要在.py文件最开头添加你工具包的路径
import sys
sys.path.append("/home/lh/.conda/envs/pytorchcpu/lib/python3.9/site-packages")
打个小广告
有想学习深度学习pytorch框架的朋友,可以参考本人相关仓库