Skip to content

Taichi入门

???+ note "前言" Taichi 是一个基于 Python 的领域特定语言,专为高性能并行计算设计。

安装

bash
pip install taichi

在pycharm中新建一个Pycharm项目

![](1 copy.png) ???+ note "注意" 这里选择 python Interpreter 时需要选择python的默认安装目录中的python.exe,不然会导致已经下载的第三方库(比如taichi)在pycharm中无法使用

开始写Taichi代码

taichi本来是用来做计算机图形学的内容,由于笔者目前对计算机图形学并不太了解,这里简单演示一下如何用taichi来绘制一条直线

导入taichi模块

python
import taichi as ti

初始化taichi

python
ti.init(arch = ti.gpu)

此行调用 ti.init(),根据传入参数定制 Taichi 运行时。 目前,我们只介绍最重要的参数:arch

参数 arch 指定后端来执行已编译的代码。 后端可以是 ti.cputi.gpu。 当指定后端为 ti.gpu 时,Taichi 按照 ti.cudati.vulkanti.opengl/ti.metal 的优先顺序选择后端。 如果没有 GPU 架构可用,Taichi 回退到 CPU 设备。

你也可以直接指定要使用的 GPU 后端。 例如,设置 arch=ti.cuda,在 CUDA 上运行程序。 如果目标架构不可用,Taichi 会报错。

定义一段直线

python
import numpy as np
X = np.random.random((1, 2)) #x 为起点
Y = np.random.random((1, 2)) #y 为终点

在taichi提供的图像界面上显示

python
n = 560
gui = ti.gui("我的第一个taichi作品",res = (n*2,n))
for i in range(10000):
    gui.lines(begin=X, end=Y, radius=2, color=0x068587)
    gui.show()

???+ info "注意" 为了防止图形界面一闪而过,我们还要加入以下代码 python while gui.running: gui.show()