pycuda安装与使用

1.安装

（1）找到适合自己Python解释器和CUDA的pycuda版本
在这个网站上下载网站找到Pycuda

网站首页

找到pycuda

如：pycuda‑2020.1+cuda101‑cp37‑cp37m‑win_amd64.whl
pycuda‑2020.1表示pycuda版本为 2020.1
cuda101表示对应cuda版本为10.1
cp37表示Python解释器版本为3.7
找到对应的版本即可，这里可以不下载，用pip命令下载安装

（2）安装

pip install pycuda == 2020.1
这样就装好了!
**注意：**一定对应好自己的CUDA和Python版本
我的是python3.7.6和CUDA10.1

2. 简单使用

import pycuda.autoinit
import pycuda.driver as cuda
from pycuda.compiler import SourceModule
import numpy as np
import time# 核函数编写
mod = SourceModule("""
自己的核函数
""")函数名 = mod.get_function("自己写的核函数名")if __name__ == '__main__':# 设置矩阵规格n = 10000block_size = 16 # 设置grid和blockblock = (block_size, block_size, 1)if n % block_size != 0:grid = (n // block_size + 1, n // block_size + 1, 1)else:grid = (n / /block_size, n // block_size, 1)# 执行GPU程序函数名（参数）

我的是pycuda的矩阵乘法
运行成功啦
10000 * 10000的矩阵乘法

pycuda安装与使用相关推荐

pycuda安装报错（已解决）
在安装pycuda的过程中,出现找不到cuda.h文件,出现这种问题有俩种原因: 1.确实没有找到cuda.h文件,原因是环境变量没有包含cuda路径,打开~/.bashrc添加环境路径: expor ...
TensorRT及PyCUDA安装记录
一.PyCUDA安装记录环境版本说明: Ubuntu 16.04 LTS python3.6 CUDA11.0 cuDNN8.0 Pytorch 1.7 Tensorflow(GPU) 2.4.0 ...
windows 下 pycuda 安装问题
ERROR: Failed building wheel for pycuda Failed to build pycuda ERROR: Could not build wheels for pyc ...
TRT使用之pycuda安装
执行:pip install pycuda 出现:Building wheel for pycuda (PEP 517) ... error 使用:pip install pycuda --no-bi ...
pycuda 安装完毕，验证步骤
pycuda 按理说直接pip安装就可,但是如果你遇到问题,要么从源码编译安装,windows的话,可以从: https://www.lfd.uci.edu/~gohlke/pythonlibs/#p ...
双硬盘安装双系统，配置显卡驱动，安装cuda,pycuda
本文主要记录了在双硬盘笔记本电脑自带的Windows10之外安装ubuntu18.04,并安装NVIDIA显卡驱动与cuda 和 pycuda 的步骤和注意事项. 软硬件清单: 128G SSD + ...
pycuda的安装，wheel报错
pycuda是调用显卡的的一个包对于并行计算和人工智能图像分析都有很大的帮助但是网上的pycuda安装都是不太好用所以我给大家来介绍一下我怎么安装成功的安装cuda (pip install ...
ubuntu安装pyCUDA
0. 写在前面安装环境:ubuntu18.04(16和18差不多,但是18太爽了)和python3(具体版本忘了,应该是3.6) 参考链接:参考了验证程序 1. 安装pyCUDA之前必须安装CUDA ...
pycuda的安装和错误调试
计算机各配置版本信息: cuda:9.0 cudnn:7.4.2 python:3.5 pycuda安装: pip install pycuda==2017.1.1 安装后运行报错:FileNotFo ...

pycuda安装与使用

pycuda安装与使用

1.安装

网站首页

找到pycuda

2. 简单使用

pycuda安装与使用相关推荐

最新文章

热门文章