python3.9 => python

PennyLaneAI · Oct 12, 2023 · 641be3b · 641be3b
1 parent 610a273
commit 641be3b
Showing 1 changed file with 4 additions and 5 deletions.
diff --git a/.github/workflows/wheel_linux_x86_64_cu11.yml b/.github/workflows/wheel_linux_x86_64_cu11.yml
@@ -65,7 +65,7 @@ jobs:
           echo "COMPILER=g++" >> $GITHUB_ENV
 
       - name: Install cibuildwheel
-        run: python3.9 -m pip install cibuildwheel~=2.11.0
+        run: python -m pip install cibuildwheel~=2.11.0
 
       - name: Build wheels
         env:
@@ -81,19 +81,18 @@ jobs:
           # Python build settings
           CIBW_BEFORE_BUILD: |
             cat /etc/yum.conf | sed "s/\[main\]/\[main\]\ntimeout=5/g" > /etc/yum.conf
-            python -m pip install ninja cmake~=3.24.3 auditwheel
-            python -m pip install custatevec-cu11
-            yum-config-manager --add-repo https://developer.download.nvidia.com/compute/cuda/repos/rhel7/x86_64/cuda-rhel7.repo -y
+            python -m pip install ninja cmake~=3.24.3 auditwheel custatevec-cu11
             yum clean all -y
             yum install centos-release-scl-rh -y
             yum install devtoolset-11-gcc-c++ -y
             source /opt/rh/devtoolset-11/enable -y
+            yum-config-manager --add-repo https://developer.download.nvidia.com/compute/cuda/repos/rhel7/x86_64/cuda-rhel7.repo -y
             yum -y install cuda-11-5 git openssh wget
 
           # ensure nvcc is available
           CIBW_ENVIRONMENT: |
             PATH=$PATH:/usr/local/cuda/bin \
-            LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda/lib64:$(python -c "import site; print( f'{site.getsitepackages()[0]}/cuquantum/lib')") \
+            CUQUANTUM_SDK=$(python -c "import site; print( f'{site.getsitepackages()[0]}/cuquantum/lib')") \
             PL_BACKEND="${{ matrix.pl_backend }}"
 
           # CIBW_MANYLINUX_X86_64_IMAGE: manylinux2014