StateTomography - kernel crashes for StateTomography()

[giladqm]

Hi,

I have a custom backend that simulates a real quantum chip with 21 qubits. My experiment is a simple 5 qubit circuit that does teleportation from the 1st qubit to the 3rd to the 5th.

I want to get the density matrix of the qubit after teleportation and also represent the original state using the Bloch sphere.

For now I'm using StateTomography() with my circuit before transpilation on my backend, but when I try to run the experiment with the transpiled circuit (which is optimized) my kernel crashes :(

I'm using a very strong computer for this (Grace-Hopper from nvidia), and I'm running a notebook (ipynb) in a cuquantum container.

Moreover, is there a way to utilize my GPU for this?

Thanks!