简易机器学习笔记(十)Windows下 PaddlePaddle配置CUDA加速环境

本文指导读者如何为设备安装CUDA12.0版本及cuDNNv8.9.1,以配合PaddlepaddleGPU版,包括检查CUDA版本、安装步骤、cuDNN文件夹放置和测试环境设置。
摘要由CSDN通过智能技术生成

前言

大伙既然都来做这个了,那配个CUDA环境肯定是必不可少的了吧(笑)

最前面的最前面,

流程

  1. 确定当前设备支持的CUDA版本
  2. 安装CUDA Toolkit 和 GPU版的Paddlepaddle
  3. 下载cuDNN Archive
  4. 手动放置配套的cuDNN到指定文件夹
  5. 测试

1.确定当前设备支持的CUDA版本

当然了,我这里默认了你是有安装NVIDIA驱动的,如果没有,你可以自己百度一下如何安装NVIDIA驱动,或者直接下载一个Geforce experience来更新驱动。

首先右键桌面,点击NVIDIA控制面板

在这里插入图片描述
在NVIDIA控制面板下面,点击系统信息

在这里插入图片描述

下图中点击组件
在这里插入图片描述
可以在3D设置中看到自己当前支持的CUDA版本,比如我这里是12.3.107

在这里插入图片描述

2.安装CUDA Toolkit

在安装前,我希望你确定一件事,这里我们找到paddlepaddle官方网站的安装界面,这里是有这么一句话:
在这里插入图片描述

我这里最高能用CUDA 12.0版本,那么我们就需要配合cuDNN v8.9.1版本使用,还需要搭载TensorRT 8.6.1.6

我先说明,你在使用paddle的时候,请先下载最基本的CPU版本,然后再安装GPU版本,貌似GPU版本是CPU版本的一个插件一样而不是一个单独可以使用的库。

至于安装命令,可以参考

paddle官网

这里由于是12.0的CUDA版本,所以我这里命令是
在这里插入图片描述

python -m pip install paddlepaddle-gpu==2.6.0.post120 -f https://www.paddlepaddle.org.cn/whl/windows/mkl/avx/stable.html

其次,下载CUDA的链接如下:

CUDA Toolkit Archive

在此链接中找到我们心仪的12.0版本,进行下载
在这里插入图片描述
在这里插入图片描述

CUDA ToolKit就随便安装一下就可以了,直接一直点下一步即可。

3.下载cuDNN Archive

还记得吗,因为我们的CUDA版本为12.0,所以我们现在需要下载的cuDNN版本为v8.9.1

现在找到cuDNN的下载地址:

cuDNN Archive

这里需要登陆NVIDIA官方,可能需要梯子

在这里插入图片描述
这里因为我们是12.x版本,所以就下载上面这个cuDNN版本

4. 手动放置配套的cuDNN到指定文件夹

ok我们现在下好了这个cuDNN包,然后里面是这些东西
在这里插入图片描述

这几个文件夹待会要用,现在先找到环境变量,找到这几项:;
在这里插入图片描述

因为我们是12.0的版本,所以我们转到这个12.0的目录,然后把cuDNN包里面这几个文件夹直接复制进去

在这里插入图片描述

5.测试吧!

测试只需要输入代码如下即可:

import paddle

paddle.utils.run_check()

 # 开启0号GPU训练
use_gpu = True
paddle.device.set_device('gpu:0') if use_gpu else paddle.device.set_device('cpu')

如果看到代码没报错,有如下字样,则代表设备初始化成功,库的导入成功啦!

在这里插入图片描述

  • 20
    点赞
  • 24
    收藏
    觉得还不错? 一键收藏
  • 1
    评论
Windows 10上配置机器学习环境可以按照以下步骤进行: 1. 安装Python环境:首先,你需要安装Python。你可以从Python官方网站下载并安装最新版本的Python。另外,你也可以选择安装Anaconda,它是一个Python发行版,内置了许多常用的机器学习库和工具。在安装Anaconda时,你可以选择安装Python 3.6版本,因为很多机器学习库都适配了这个版本[2]。 2. 安装CUDA:如果你计划使用GPU加速机器学习训练过程,你需要安装CUDACUDA是NVIDIA提供的用于GPU计算的平台和编程模型。你可以从NVIDIA官方网站下载并安装适合你的显卡型号的CUDA版本。 3. 安装IDE:选择一个适合你的集成开发环境(IDE)来编写和运行机器学习代码。常用的IDE包括PyCharm、Visual Studio Code和Jupyter Notebook等。你可以根据个人喜好选择一个合适的IDE。 4. 安装微信开发者工具:如果你计划开发机器学习微信小程序,你需要安装微信开发者工具。微信开发者工具提供了开发、调试和发布微信小程序的功能。你可以从微信官方网站下载并安装微信开发者工具。 此外,为了更好地管理不同版本的Python和依赖包,你可以考虑使用Python虚拟环境。虚拟环境可以帮助你隔离不同项目的依赖关系,避免版本冲突[3]。 总结起来,配置Windows 10上的机器学习环境包括安装Python环境、安装CUDA(如果需要GPU加速)、选择合适的IDE以及安装微信开发者工具。另外,你还可以考虑使用Python虚拟环境来管理不同项目的依赖关系。

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值