PyCUDA

发布时间：2023-12-14 05:58:18 所属栏目：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() # 查询当前显卡的总显存，可用显存文档

官方文档

（编辑：东莞站长网）

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