admin管理员组

文章数量:1530018

由于笔记本显卡是AMD的集显,没有NVIDIA的独立显卡,所以在下载pytorch版本的时候应该选择cpu版本

pytorch下载链接

选择如下图所示方案进行pytorch的下载;

将以下代码在cmd中运行,进行pytorch的安装

pip3 install torch torchvision torchaudio

因为后期需要用到yolov5,需要保证torch版本>1.7.0,可以使用以下命令:

pip3 install torch>=1.7.0 torchvision torchaudio


成功安装!

本文标签: 方案amdPytorch