Add cuda device id

8ff · Sep 15, 2023 · 3968644 · 3968644
1 parent e3d05c0
commit 3968644
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/pkg/objectPredict/objectPredict.go b/pkg/objectPredict/objectPredict.go
@@ -227,7 +227,7 @@ func (c *Client) initSession() (ModelSession, error) {
 		defer cudaOptions.Destroy()
 
 		// This is a clunky API, but it reflects how the underlying C API sets CUDA options.
-		err = cudaOptions.Update(map[string]string{"device_id": string(c.CudaDeviceID)})
+		err = cudaOptions.Update(map[string]string{"device_id": fmt.Sprintf("%d", c.CudaDeviceID)})
 		if err != nil {
 			return ModelSession{}, fmt.Errorf("error updating CUDA provider options: %w", err)
 		}