CUDA——Python基础与实现

CUDA是NVIDIA针对GPU加速计算而开发的编程框架。通过PyTorch可以很方便地使用CUDA，无须过多地改变代码。

一、numpy运算和python运算

使用 numpy 进行矩阵乘法比使用 Python 的效能会提高很多。

import torch
import numpy# 方形矩阵大小
size = 600a = numpy.random.rand(size, size)
b = numpy.random.rand(size, size)

使用numpy将两个数列相乘，使用timeeit进行计时：

%%timeitx = numpy.dot(a,b)

直接使用python相乘：

%%timeitc = numpy.zeros((size,size))for i in range(size):for j in range(size):for k in range(size):c[i,j] += a[i,k] * b[k,j]passpass

比较二者的用时差异：

第二个用时到达了3分钟多，没有截屏。

二、GPU上的torch张量

创建一个 GPU 中的张量，并检查数据格式。

x = torch.cuda.FloatTensor([3.5])
x.type()
x.device

三、试验GPU性能

# 将前面的numpy数据转换为cuda格式aa = torch.cuda.FloatTensor(a)
bb = torch.cuda.FloatTensor(b)

开始运算：

%%timeitcc = torch.matmul(aa, bb)

比之前还快。

四、检查CUDA是否可用的标准代码

# 检查CUDA是否可用
# 如果可用，转换为cuda格式if torch.cuda.is_available():torch.set_default_tensor_type(torch.cuda.FloatTensor)print("using cuda:", torch.cuda.get_device_name(0))passdevice = torch.device("cuda" if torch.cuda.is_available() else "cpu")device

CUDA——Python基础与实现相关推荐

CUDA PYTHON 并行计算基础
一.CUDA异构计算基础 1.CUDA简介 CUDA(Compute Unified Device Architecture),是一种基于C/C++的编程方法,支持异构编程的扩展方法,提供了简单明了的 ...
python基础知识及数据分析工具安装及简单使用(Numpy/Scipy/Matplotlib/Pandas/StatsModels/Scikit-Learn/Keras/Gensim))
Python介绍. Unix & Linux & Window & Mac 平台安装更新 Python3 及VSCode下Python环境配置配置 python基础知识及数据分 ...
五节课从零起步(无需数学和Python基础)编码实现AI人工智能框架电子书V1
五节课从零起步 (无需数学和Python 基础) 编码实现AI 人工智能框架王家林 2018/4/15 ...
f是一个python内部变量类型,Python基础变量类型——List浅析
Python使用list 一.list Python内置的一种数据类型是列表:list.list是一种有序的集合,可以随时添加和删除其中的元素. 比如,列出班里所有同学的名字,就可以用一个list表示 ...
python 等号报错_Python学习----Python基础
Python基础一.数据类型和变量 1.在Python中,等号=是赋值语句,可以把任意数据类型赋值给变量,同一个变量可以反复赋值,而且可以是不同类型的变量. 例如: a =520# a是整数prin ...
python基础——继承与派生、组合
python基础--继承与派生 1 什么是继承: 继承是一种创建新的类的方式,在python中,新建的类可以继承自一个或者多个父类,原始类成为基类或超累,新建的类成为派生类或子类 1.1 继承分为:单 ...
Python学习笔记1 Python基础
第1章 Python基础 1.1 Python概述及版本介绍 1.Python是一种面向对象的解释型计算机程序设计语言(解释型语言源代码->解释器逐行解释每一句源代码) 2.优点:高级语言.可移 ...
python基础---元组、字典、函数、文件、异常
文章目录 python基础---元组.字典.函数.文件.异常 Tuple(元组) 常用操作 dict(字典) 函数文件异常 python基础-元组.字典.函数.文件.异常 Tuple(元组) tu ...
快速掌握Python的捷径-Python基础前传（1）
文: jacky(朱元禄) 开文序最近看新闻,发现高考都考Python了,随着人工智能的火热,学数据科学的人越来越多了!但对于数据行业本身来说,现象级的火热,这并不是什么好事. 方丈高楼平地起,无 ...

CUDA——Python基础与实现

一、numpy运算和python运算

二、GPU上的torch张量

三、试验GPU性能

四、检查CUDA是否可用的标准代码

CUDA——Python基础与实现相关推荐

最新文章

热门文章