diff --git a/Generator(forGPU)/gen_quantum_mp.py b/Generator(forGPU)/gen_quantum_mp.py index 1a3ce31..ee602b8 100644 --- a/Generator(forGPU)/gen_quantum_mp.py +++ b/Generator(forGPU)/gen_quantum_mp.py @@ -101,7 +101,8 @@ def _setup_simulator(self): except Exception as e: self.backend = AerSimulator(method='statevector', device='CPU') self.gpu_available = False - print(f"[CPU] Using CPU (GPU not available: {str(e)[:50]})") + error_msg = str(e)[:80] + print(f"[CPU] Using CPU (GPU not available: {error_msg})") # Qiskit 2.x: Use StatevectorSampler (no backend parameter needed) self.sampler = StatevectorSampler() @@ -597,7 +598,7 @@ def worker_generate_sample(args): """Worker function for multiprocessing""" sample_id, n_nodes, edge_prob, problem_type, p, seed, use_gpu = args - gen = MultiProblemQuantumGenerator(seed=seed, use_gpu=False, n_workers=1) + gen = MultiProblemQuantumGenerator(seed=seed, use_gpu=use_gpu, n_workers=1) sample = gen.generate_single_sample(sample_id, n_nodes, edge_prob, problem_type, p, seed) if sample: diff --git a/Generator(forGPU)/requirements.txt b/Generator(forGPU)/requirements.txt new file mode 100644 index 0000000..5d9d09b --- /dev/null +++ b/Generator(forGPU)/requirements.txt @@ -0,0 +1,18 @@ +# Core dependencies +numpy>=1.21.0 +networkx>=2.6 +scipy>=1.7.0 + +# Qiskit quantum computing framework +qiskit>=1.0.0 +qiskit-algorithms>=0.3.0 +qiskit-aer>=0.14.0 + +# GPU support (optional but recommended for performance) +# Uncomment the line below if you have NVIDIA GPU with CUDA support +# qiskit-aer-gpu>=0.14.0 + +# Development tools (optional) +# pytest>=7.0.0 +# black>=22.0.0 +# pylint>=2.10.0