加入收藏 | 设为首页 | 会员中心 | 我要投稿 阜阳站长网 (https://www.0558zz.com/)- 科技、建站、内容创作、云计算、网络安全!
当前位置: 首页 > 综合聚焦 > Ubuntu > 正文

PyCUDA

发布时间:2023-12-19 06:16:50 所属栏目:Ubuntu 来源:DaWei
导读: PyCUDA - Ubuntu14.04安装及测试
安装

依赖 Numpy
直接安装

sudo apt install python-pycuda

源码安装:

Step 1 下载并解压 PyCUDA
$ wget http://pypi.python.org/pypi

PyCUDA - Ubuntu14.04安装及测试 安装 依赖 Numpy 直接安装 sudo apt install python-pycuda 源码安装: Step 1 下载并解压 PyCUDA $ wget http://pypi.python.org/pypi/pycuda $ tar xfz pycuda-VERSION.tar.gz Step 2 编译PyCUDA $ cd pycuda-VERSION # if you're not there already $ python configure.py --cuda-root=/where/ever/you/installed/cuda $ su -c "make install" 测试 e.g. 1 $ cd pycuda-VERSION/test $ python test_driver.py

输出 OK 即成功.

e.g. 2 $ cd pycuda-VERSION/examples $ python test_demo.py e.g. 3 import pycuda.autoinit import pycuda.driver free_bytes,total_bytes = pycuda.driver.mem_get_info() # 查询当前显卡的总显存,可用显存 文档

官方文档

(编辑:阜阳站长网)

【声明】本站内容均来自网络,其相关言论仅代表作者个人观点,不代表本站立场。若无意侵犯到您的权利,请及时与联系站长删除相关内容!

    推荐文章