yolov8环境配置---安装pytorch

一，前面步骤请参考上一篇博客

yolov8环境配置

二，安装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版本安装命令如下

四，等待安装完成即可使用