yolov8环境配置---安装pytorch

2024-01-09 22:29:27

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

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

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

文章来源:https://blog.csdn.net/sanqianyanyue/article/details/135465673
本文来自互联网用户投稿，该文观点仅代表作者本人，不代表本站立场。本站仅提供信息存储空间服务，不拥有所有权，不承担相关法律责任。