[急！]使用taichi.init的时候怎么指定使用哪块gpu？

综合讨论区
#1

大佬们急求解答，默认的ti.init(arch=ti.gpu, device_memory_fraction=0.3)只会使用第一块gpu，怎么指定使用别的gpu呢

#2

你好:wave: 请问你的机器是什么系统， GPU 都是什么卡？

#3

linux ubuntu18.04系统服务器, GPU都是英伟达RTX3090

#4

CUDA 的话可以用一些方法来指定 primary GPU：cuda - How do I select which GPU to run a job on? - Stack Overflow

Taichi 里据我所知没有指定 GPU 的方法，这个任务应该是交给 CUDA 后端去处理的。

#5

感谢大佬！俺去瞅瞅