一,前面步骤请参考上一篇博客
二,安装pytorch---cpu版本
1,进入官网根据自己电脑选择安装GPU或CPU版本,有独立显卡可安装GPU(运行的时候速度快)pytorch
2,安装cpu版本(要求pytorch>=1.8),选择所需要的版本复制命令,到自己所创建yolo虚拟环境中运行即可
上图中为最新版本,可以选择历史版本如下(再其中选择需要的命令在环境中运行即可)
三,pytorch安装---gpu版本
1,cpu版本需要查看电脑cuda版本(这里有两个方法)
方法一:在Anaconda Prompt (Anaconda)中输入命令:nvidia-smi
nvidia-smi
其中右上角CUDA Version:11.7即为当前电脑CUDA版本
方法二:打开NIVIDIA
点击系统信息后,点击组件,如图横线所示即为当前电脑CUDA版本
![]()
2,进入官网,选择安装所需要的版本,复制命令运行即可安装
注:
1,gpu版本要选择要与电脑CUDA版本相匹配
2,pip速度较快
3,选择历史版本与CUDA版本安装命令如下
四,等待安装完成即可使用