详细介绍如何使用GPU(显卡)跑MATLAB代码

一、概述

想要在GPU跑MATLAB,首先得查看你的电脑是否支持了。

先来了解一个概念——CUDA:是显卡厂商NVIDIA推出的运算平台。 CUDA是一种由NVIDIA推出的通用并行计算架构,该架构使GPU能够解决复杂的计算问题。

说白了,就是你用显卡进行深度学习等的运算必须要安装cuda包(你可以了理解为一个软件),用GPU跑MATLAB代码也必须要用到cuda包,每一个显卡只能安装对应版本的cuda包。

二、查看cuda版本并安装

依次打开:控制面板==>NVIDIA控制面板==>帮助==>系统信息==>组件==>NVCUDA.DLL==>产品名称

可以看到我的电脑支持cuda 10.1版本

 此时就需要在你电脑安装对应版本的cuda版本,可以查看往期教程:

tips:如果有深度学习相关的cudnn也建议安装

 全网最详细的深度学习tensorflow-gpu环境配置_惜洛-Jankin的博客-CSDN博客

……

三、确认MATLAB版本与cuda版本对应

安装好之后需要确认MATLAB版本与cuda版本对应才行,比如我的cuda10.1,只能使用MATLAB2019b与2020a

GPU Support by Release- MATLAB & Simulink- MathWorks 中国

四、在MATLAB查看GPU版本

到此你已经安装好了cuda了,MATLAB版本也对,就可以打开MATLAB查看GPU了,若是你没有安装cuda,会提示错误的。

在命令行输入

gpuDevice

正常情况会出现:

五、测试gpu跑代码 

使用gpu跑代码首先需要了解几个函数

gpuArray存储在 GPU 上的数组
gather将分布式数组或 gpuArray 传输到本地工作空间
gpuDevice查询或选择 GPU 设备
arrayfun将函数应用于 GPU 上数组的每个元素

最主要的就是前两个了,在运算时,必须将数组等定义为gpuArray类型,代码才会在gpu运行。

若是想把gpuArray对象转为普通数组,就要用到gateer函数了。

gpuArrary支持的函数:参考列表- MATLAB & Simulink- MathWorks 中国

举例:

G = gpuArray(X)
将数组复制 X到 GPU 并返回一个gpuArray对象。
G就是一个可以在gpu运行的数组

测试速度

输入如下代码在cpu运行并显示时间

tic
G=randn(1e4);
toc

输入如下代码在gpu运行并显示时间

tic
G=randn(1e4,'gpuArray');
toc

 还是快了不少。

  • 22
    点赞
  • 187
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 9
    评论
以下是使用GMRES GPU算法的MATLAB代码示例: ```matlab %GMRES-GPU algorithm %The matrix A and the vector b are stored in the device (GPU) %The initial guess x0 is also stored in the device (GPU) %The algorithm solves Ax = b for x function [x, flag, relres, iter] = gmres_gpu(A, b, x0, rtol, maxit) %Transfer A and b from the host (CPU) to the device (GPU) A_gpu = gpuArray(A); b_gpu = gpuArray(b); %Initialize variables n = size(A, 1); x = gpuArray(x0); r = b_gpu - A_gpu * x; normb = norm(b_gpu); normr = norm(r); %Check for initial guess x0 if normr/normb <= rtol flag = 0; relres = normr/normb; iter = 0; return; end %Initialize GMRES variables m = 30; %maximum number of iterations Q = zeros(n, m+1, 'gpuArray'); H = zeros(m+1, m, 'gpuArray'); cs = zeros(m, 1, 'gpuArray'); sn = zeros(m, 1, 'gpuArray'); e1 = [1; zeros(m, 1), 'gpuArray']; Q(:,1) = r/normr; beta = normr; j = 0; while normr/normb > rtol && j < maxit j = j + 1; %Arnoldi process for i = 1:m v = A_gpu * Q(:,i); for k = 1:i H(k,i) = Q(:,k)' * v; v = v - H(k,i) * Q(:,k); end H(i+1,i) = norm(v); Q(:,i+1) = v / H(i+1,i); %Apply Givens rotations for k = 1:i-1 temp = cs(k) * H(k,i) + sn(k) * H(k+1,i); H(k+1,i) = -sn(k) * H(k,i) + cs(k) * H(k+1,i); H(k,i) = temp; end %Compute Givens rotation [cs(i), sn(i), H(i,i)] = givens(H(i,i), H(i+1,i)); H(i+1,i) = 0; %Apply Givens rotation to RHS temp = cs(i) * beta; beta = -sn(i) * beta; beta = cs(i) * (H(i,i) - temp) + sn(i) * H(i+1,i); H(i+1,i) = -sn(i) * (H(i,i) - temp) + cs(i) * H(i+1,i); %Check for convergence if abs(beta) <= rtol y = H(1:i,1:i) \ beta * e1(1:i); x = x + Q(:,1:i) * y; relres = norm(A_gpu * x - b_gpu) / normb; flag = 0; iter = j; return; end end %Solve least squares problem y = H(1:m,1:m) \ beta * e1(1:m); x = x + Q(:,1:m) * y; %Compute residual r = b_gpu - A_gpu * x; normr = norm(r); %Check for convergence if normr/normb <= rtol relres = normr/normb; flag = 0; iter = j; return; end end %Check for maximum number of iterations if j >= maxit flag = 1; else flag = -1; end %Compute relative residual relres = normr/normb; %Return number of iterations iter = j; end %Givens rotation function function [c, s, r] = givens(a, b) if b == 0 c = sign(a); s = 0; r = abs(a); elseif abs(b) > abs(a) temp = a / b; s = sign(b) / sqrt(1 + temp^2); c = s * temp; r = b / s; else temp = b / a; c = sign(a) / sqrt(1 + temp^2); s = c * temp; r = a / c; end end ``` 请注意,此代码需要GPU计算能力。您还需要确保您的MATLAB版本支持GPU计算。

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

惜洛-Jankin

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值