たとえば、2 つの GPU と 2 つのホスト スレッドがあります。multigpu PCが遠いので確認できません。最初のホスト スレッドを最初の GPU で動作させ、2 番目のホスト スレッドを 2 番目の GPU で動作させたいと考えています。すべてのホスト スレッドは、多くの cublas 呼び出しで構成されています。cudaSetDevice() 呼び出しによって、最初のホスト スレッドから最初の GPU を選択し、2 番目のホスト スレッドから 2 番目の GPU を選択することは可能ですか?
たとえば、2 番目のホスト スレッドには を呼び出しcudaSetDevice(1)、最初のスレッドには を呼び出しますcudaSetDevice(0)。