improve dockerfile

Author: refraction-ray

.dockerignore

@@ -3,4 +3,13 @@
 **/*idea
 **/*mypy_cache
 **/*pytest_cache
-**/*develop
+**/*develop
+**/*dist
+**/*build
+**/*htmlcov
+**/*__pycache__
+**/*.pyc
+**/*.outdated
+**/*.result
+**/*.results
+**/*.data

CHANGELOG.md

@@ -12,6 +12,8 @@
 
 - type conversion in measure API when high precision is set
 
+- fix bug in to_qiskit with new version qiskit
+
 ## 0.0.220509
 
 ### Added

docker/Dockerfile

@@ -1,10 +1,21 @@
 FROM nvidia/cuda:11.0.3-cudnn8-runtime-ubuntu18.04
 
+# https://gitlab.com/nvidia/container-images/cuda/-/blob/master/doc/supported-tags.md
+
+RUN apt-key del 7fa2af80
+RUN apt-key del 3bf863cc
+RUN apt-key adv --fetch-keys https://developer.download.nvidia.com/compute/cuda/repos/ubuntu1804/x86_64/3bf863cc.pub
+RUN apt-key adv --fetch-keys https://developer.download.nvidia.com/compute/cuda/repos/ubuntu1804/x86_64/7fa2af80.pub
+
+# Nvidia is breaking the docker images on Apr 2022 ...
+# https://github.com/NVIDIA/nvidia-docker/issues/1631
+
 RUN apt update && DEBIAN_FRONTEND=noninteractive apt install -y \
     wget \
     cuda-command-line-tools-11-0 \
     git \
-    vim
+    vim \
+    pandoc
 
 RUN wget -q -P /tmp \
   https://repo.anaconda.com/miniconda/Miniconda3-latest-Linux-x86_64.sh \
@@ -26,8 +37,16 @@ RUN pip install jaxlib==0.3.2+cuda11.cudnn805 -f https://storage.googleapis.com/
 
 RUN pip install -U git+https://github.com/jcmgray/cotengra.git
 
+RUN pip install ray
+
+# requirements conflict for ray
+
 RUN ln -s /usr/local/cuda-11.0/targets/x86_64-linux/lib/libcusolver.so.10 /usr/local/cuda-11.0/targets/x86_64-linux/lib/libcusolver.so.11
 
+RUN echo export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda-11.0/targets/x86_64-linux/lib >> ~/.bashrc \
+    && echo export PYTHONPATH=/app >> ~/.bashrc \
+    && echo export TF_CPP_MIN_LOG_LEVEL=3 >> ~/.bashrc
+
 COPY . /app
 
 # RUN export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda-11.1/targets/x86_64-linux/lib

examples/bp_benchmark.py

@@ -6,10 +6,10 @@
 os.environ["TF_FORCE_GPU_ALLOW_GROWTH"] = "true"
 import time
 import tensorflow as tf
-import tensorflow_quantum as tfq
 import cirq
 import sympy
 import numpy as np
+import tensorflow_quantum as tfq
 import tensorcircuit as tc
 
 

requirements-docker.txt

@@ -1,6 +1,6 @@
 numpy
 scipy
-cirq==0.11
+cirq
 tensorflow==2.7
 tensornetwork==0.4.6
 graphviz
@@ -23,7 +23,7 @@ quimb
 openfermionpyscf
 # below is for development
 mypy==0.782
-pytest==6.2.4
+pytest
 pytest-cov
 pytest-benchmark
 pytest-xdist
@@ -34,7 +34,7 @@ sphinx-intl
 sphinx-copybutton
 nbsphinx
 furo
-pylint==2.11.1
+pylint
 ray
 pennylane
 tensorflow_quantum==0.6.1

tensorcircuit/translation.py

@@ -110,11 +110,11 @@ def qir2qiskit(qir: List[Dict[str, Any]], n: int) -> Any:
             qiskit_circ.unitary(wroot_op, index, label=qis_name)
         elif gate_name in ["rx", "ry", "rz", "crx", "cry", "crz"]:
             getattr(qiskit_circ, gate_name)(
-                np.real(backend.numpy(parameters["theta"])), *index
+                np.real(backend.numpy(parameters["theta"])).item(), *index
             )
         elif gate_name in ["orx", "ory", "orz"]:
             getattr(qiskit_circ, "c" + gate_name[1:])(
-                np.real(backend.numpy(parameters["theta"])), *index, ctrl_state=0
+                np.real(backend.numpy(parameters["theta"])).item(), *index, ctrl_state=0
             )
         elif gate_name in ["exp", "exp1"]:
             unitary = backend.numpy(backend.convert_to_tensor(parameters["unitary"]))