こんにちは。Qiskitを高速化したいですよね。バックエンドを切り替えるだけで早くなります。ちょっと環境構築には苦労するかもしれませんが。今回はblueqat cloudを利用したQiskitの簡単な高速化を紹介します。blueqat cloudは有料版の企業版を利用しています。GPUはV100の32Gです。

結果から見てみましょう。横軸は量子ビット数で22量子ビットまで。縦軸は計算にかかった時間です。CPUはIntelXeonを使ってます。

GPUの秒の雰囲気がわからないので、リストを書き出してみました。ほぼ0.1秒未満です。

[0.004900693893432617,
 0.07413864135742188,
 0.024965763092041016,
 0.024173974990844727,
 0.028377056121826172,
 0.029413223266601562,
 0.03184700012207031,
 0.03441810607910156,
 0.03339433670043945,
 0.03801369667053223,
 0.039711713790893555,
 0.04146695137023926,
 0.04342055320739746,
 0.04290771484375,
 0.045015811920166016,
 0.05161881446838379,
 0.0559999942779541,
 0.05427384376525879,
 0.06047534942626953,
 0.06277298927307129,
 0.06796908378601074,
 0.07407522201538086,
 0.0837862491607666]

計算した回路はRXとCXを組み合わせた適当な回路を10回繰り返したものです。

コードを見てみます。

from qiskit import QuantumCircuit, transpile
from qiskit import Aer
import random
import time
def create_ghz_circuit(n_qubits):
  circuit = QuantumCircuit(n_qubits)
  loop = 10
  for _ in range(loop):
    for qubit in range(n_qubits):
      circuit.rx(random.random(),qubit)
    for qubit in range(n_qubits - 1):
      circuit.cx(qubit, qubit + 1)
  return circuit

実装は本当にバックエンドを変更するだけです。aerの標準の状態ベクトルシミュレータでの計算とGPUでの計算をやってます。

arr_gpu = []
arr_cpu = []

for n_qubits in range(0,23):
  backend = Aer.get_backend('aer_simulator_statevector')
  circuit = create_ghz_circuit(n_qubits=n_qubits)
  circuit.measure_all()
  circuit = transpile(circuit, backend)
  backend = Aer.get_backend('aer_simulator')
  backend.set_options(device='GPU', cuStateVec_enable=True)
  start = time.time()
  result = backend.run(circuit).result()
  arr_gpu.append(time.time()-start)
  #print(result.get_counts())
  #print(f'backend: {result.backend_name}')
   
  backend = Aer.get_backend('aer_simulator_statevector')
  start = time.time()
  result = backend.run(circuit).result()
  arr_cpu.append(time.time()-start)
  #print(result.get_counts())
  #print(f'backend: {result.backend_name}')

計算結果の表示は、

import matplotlib.pyplot as plt

matplotlibを用いて二つの折れ線グラフを描画します

plt.plot(arr_cpu, label="CPU")
plt.plot(arr_gpu, label="GPU")
グラフのタイトルと軸のラベルを設定します

plt.xlabel('qubit')
plt.ylabel('sec')
凡例を表示します

plt.legend()
グラフを表示します

plt.show()

とにかくいいところは、今までのQiskitのソフトウェア資産を活かしながら、バックエンドの指定をcuQuantumにするだけで高速化されるということですね。最高です。数十秒、数分かかるところが0.1秒とかなので大変便利です。使いましょう！以上です。