GpuCalculateTools
GpuCalculateTools是一个基于GPU并行化技术开发的python包。它旨在提升数据的运算效率,提供了许多高效运算的算子、模块和函数等等。目前该包正在开发完善中,且仅支持英伟达系列显卡(因为笔者的显卡就是1660ti的)。后续可能会兼容更多的显卡类型。敬请期待。
版本更新浏览
- 修改了nvcc编译器的编译选项。
- 添加了对不同python版本的支持。
- 支持在conda虚拟环境中运行。
- 修改了系统路径的查询逻辑。
如何安装该程序
在安装此程序前确保您的设备安装nvcc编译器。nvcc编译器可以通过访问英伟达下载官网进行下载。下载安装后需要找到cudaToolkit的根目录,然后在系统路径配置CUDA_HOME环境变量。
如何使用GpuCalculateTools查看设备的运行参数?
通过使用get_device_prop函数可以获取显卡的运行参数,包括显卡的名称、计算能力系数、显存大小等等。该函数参数为dev_no,dev_no表示显卡的序号。如果设备中包含多个显卡,可以通过get_device_count函数获取显卡的数量,然后根据序号获取指定显卡的详细信息。
如何在指定显卡运行程序?
通过使用set_device函数指定某一个显卡运行程序。该函数参数为dev_no,dev_no表示显卡的序号。
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
File details
Details for the file gpucalculatetools-0.0.3.tar.gz.
File metadata
- Download URL: gpucalculatetools-0.0.3.tar.gz
- Upload date:
- Size: 6.4 kB
- Tags: Source
- Uploaded using Trusted Publishing? No
- Uploaded via:
twine/6.2.0 CPython/3.10.8
File hashes
| Algorithm | Hash digest | |
|---|---|---|
| SHA256 |
3641010c4836ff8b5b6b178a6205942784a0be2f8e54bc83b73aa1524574f111
|
|
| MD5 |
c338ed6f9bd18e1ec3ce5d375bb4e949
|
|
| BLAKE2b-256 |
6ee5c53b3c5cfefa1403f5d9990c859e0cfcb605d0e979c862f1d86d22255283
|