Merge pull request #588 from qiboteam/rmq2cupy

testing distributed circuits with cupy
qiboteam · May 23, 2022 · b13c7c6 · b13c7c6
2 parents 16ffea1 + 076037c
commit b13c7c6
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/.github/workflows/rules.yml b/.github/workflows/rules.yml
@@ -23,6 +23,16 @@ jobs:
         pip install .[tests,docs]
         pip install git+https://github.com/qiboteam/qibojit
         pip install tensorflow
+    - name: Install cupy on linux
+      if: startsWith(matrix.os, 'ubuntu')
+      run: |
+        sudo apt-key adv --fetch-keys https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2004/x86_64/3bf863cc.pub
+        wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2004/x86_64/cuda-ubuntu2004.pin
+        sudo mv cuda-ubuntu2004.pin /etc/apt/preferences.d/cuda-repository-pin-600
+        sudo add-apt-repository "deb https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2004/x86_64/ /"
+        sudo apt-get update
+        sudo apt install cuda-11-6
+        pip install cupy-cuda116
     - name: Test with pylint
       run: |
         pip install pylint