一,前面步骤请参考上一篇博客
二,安装pytorch---cpu版本
1,进入官网根据自己电脑选择安装GPU或CPU版本,有独立显卡可安装GPU(运行的时候速度快)pytorch
2,安装cpu版本(要求pytorch>=1.8),选择所需要的版本复制命令,到自己所创建yolo虚拟环境中运行即可
上图中为最新版本,可以选择历史版本如下(再其中选择需要的命令在环境中运行即可)
三,pytorch安装---gpu版本
1,cpu版本需要查看电脑cuda版本
一,前面步骤请参考上一篇博客
二,安装pytorch---cpu版本
1,进入官网根据自己电脑选择安装GPU或CPU版本,有独立显卡可安装GPU(运行的时候速度快)pytorch
2,安装cpu版本(要求pytorch>=1.8),选择所需要的版本复制命令,到自己所创建yolo虚拟环境中运行即可
上图中为最新版本,可以选择历史版本如下(再其中选择需要的命令在环境中运行即可)
三,pytorch安装---gpu版本
1,cpu版本需要查看电脑cuda版本