手把手教你安装CUDA(一看就会)

1.背景

学习深度学习的话,肯定需要安装PyTorch和TensorFlow,安装这两个深度学习框架之前得安装CUDA.
CUDA是什么?
CUDA是一个并行计算平台和编程模型,能够使得使用GPU进行通用计算变得简单和优雅。Nvidia官方提供的CUDA 库是一个完整的工具安装包,其中提供了 Nvidia驱动程序、开发 CUDA 程序相关的开发工具包等可供安装的选项。

2.安装过程

(1)选择安装版本:
首先打开自己电脑上的NVIDIA控制面板,点击系统信息
在这里插入图片描述
点击"组件"
在这里插入图片描述

这里显示的就是本机的版本,我这里是11.1.大部分电脑都可以向下兼容,所以也可以安装版本小于11.1的CUDA
在这里插入图片描述

(2) 到CUDA官网,点击这里
在这里插入图片描述
这里我选择的是10.1版本的CUDA
在这里插入图片描述
(3)根据自己情况进行选择,然后点击下载
在这里插入图片描述
(4)下载完成时,以管理员身份运行
在这里插入图片描述
(5)点击OK
在这里插入图片描述
(6)等待安装
在这里插入图片描述
(7)等待ing,兼容性检查
在这里插入图片描述
(8)点击同意
在这里插入图片描述
(9)选择自定义
在这里插入图片描述
(10)点击下一步
在这里插入图片描述
(11)新建文件夹,类似这种
在这里插入图片描述
在这里插入图片描述
在这里插入图片描述
(12)然后选择安装位置,前两个选择CUDA1,最后一个选择CUDA2,然后点击下一步
在这里插入图片描述
(13)勾选之后,然后点击Next
在这里插入图片描述
(14)等待安装
在这里插入图片描述
(15)点击下一步
在这里插入图片描述
(16)点击关闭
在这里插入图片描述
(17)检验是否安装成功
win + R,输入cmd打开Windows终端
输入

nvcc -V

在这里插入图片描述

附录:

参考自:B站"有幸遇见的上上"的视频

  • 100
    点赞
  • 445
    收藏
    觉得还不错? 一键收藏
  • 17
    评论
当然可以!以下是手把手你搭建 YOLOv5 的步骤: 1. 准备工作: - 确保你的电脑已经安装Python 3.8 或更高版本。 - 安装 PyTorch:可以根据你的操作系统和 CUDA 版本选择合适的安装命令,例如: ```shell pip install torch torchvision torchaudio ``` - 安装其他依赖库: ```shell pip install opencv-python matplotlib numpy ``` 2. 下载 YOLOv5 源代码: - 在终端中运行以下命令来克隆 YOLOv5 仓库: ```shell git clone https://github.com/ultralytics/yolov5.git ``` 3. 安装依赖库: - 进入 yolov5 目录: ```shell cd yolov5 ``` - 安装依赖库: ```shell pip install -r requirements.txt ``` 4. 准备数据集: - 准备好你的训练数据集,并将图像和对应的标签文件放入一个文件夹中。 5. 配置模型和数据: - 编辑 `yolov5/models/yolov5s.yaml` 文件,根据你的需求进行配置,例如设置 `nc` 参数为你的类别数量。 - 编辑 `data.yaml` 文件,设置 `train` 和 `val` 的路径,并配置类别数量和类别名称。 6. 训练模型: - 在终端中运行以下命令来开始训练模型: ```shell python train.py --img 640 --batch 16 --epochs 100 --data data.yaml --cfg models/yolov5s.yaml --weights '' ``` - 这个命令将使用默认参数来训练 YOLOv5 模型,你可以根据需要进行调整。 7. 测试模型: - 在终端中运行以下命令来测试模型效果: ```shell python detect.py --source your_image.jpg --weights runs/train/exp/weights/best.pt --conf 0.4 ``` - 这个命令将使用训练好的模型对指定图像进行目标检测,你可以调整 `--conf` 参数来控制检测结果的置信度阈值。 以上就是搭建 YOLOv5 的基本步骤,希望对你有帮助!如果有任何问题,请随时提问。

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 17
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值