具體內容看原主知乎博客:
https://zhuanlan.zhihu.com/p/97700605
最近那個99行代碼寫冰雪奇緣很火,有點好奇,來玩一玩
首先安裝好cuda10.1或者cuda10.0,我裝的是cuda10.1
裝cuda有很多教程,比較麻煩,首先得禁用nouveau驅動,安裝顯卡驅動,安裝cuda,安裝cudnn,可以參考我之前的博客
使用國內鏡像源下載,速度很快:
pip3 install --index-url http://pypi.douban.com/simple --trusted-host pypi.douban.com taichi-nightly-cuda-10-1
輸入python3
>>> import taichi as ti
[Release mode]
[T 01/10/20 17:13:32.023] [logging.cpp:Logger@68] Taichi core started. Thread ID = 6471
[Taichi version 0.3.20, cuda 10.1, commit 1c85d8e1]
>>>
安裝成功
複製代碼,復現效果: