From c3206fc567ab25db4c9906d46474cc4086ef67ca Mon Sep 17 00:00:00 2001
From: Romain Moyard <rmoyard@gmail.com>
Date: Wed, 3 May 2023 15:44:09 -0400
Subject: [PATCH] Change Torch version for GPU tests (#4080)

* Change gpu version

* Trigger CI

* Try remove 3.9 3.10

* Update cuda and add back 3.9 and 3.10

* pip install lit first

* lit no cache pip

* Setup tools

* Paths update

* Clean up

* Update

* Missing closure

* No cache dir

* Purge cache

* Formatting
---
 .github/workflows/tests-gpu.yml | 7 ++++---
 1 file changed, 4 insertions(+), 3 deletions(-)

diff --git a/.github/workflows/tests-gpu.yml b/.github/workflows/tests-gpu.yml
index ddbd19f7711..332e383119f 100644
--- a/.github/workflows/tests-gpu.yml
+++ b/.github/workflows/tests-gpu.yml
@@ -15,7 +15,7 @@ concurrency:
   cancel-in-progress: true
 
 env:
-  TORCH_VERSION: 1.13.0
+  TORCH_VERSION: 2.0.0
 
 jobs:
   gpu-tests:
@@ -69,7 +69,7 @@ jobs:
       - name: Setup Python virtual environment
         id: setup_venv
         env:
-          VENV_NAME: venv_${{ steps.setup_python.outputs.python-version }}_${{ github.sha }}
+          VENV_NAME: ${{ github.workspace }}/venv_${{ steps.setup_python.outputs.python-version }}_${{ github.sha }}
         run: |
           # Clear any pre-existing venvs
           rm -rf venv_*
@@ -97,7 +97,7 @@ jobs:
 
       - name: Install Torch
         run: |
-          python -m pip install torch==${{ env.TORCH_VERSION }} -f https://download.pytorch.org/whl/cu113/torch_stable.html
+          python -m pip install --no-cache-dir torch==${{ env.TORCH_VERSION }} -f https://download.pytorch.org/whl/cu118/torch_stable.html
 
       - name: Install PennyLane
         run: |
@@ -120,3 +120,4 @@ jobs:
           rm -rf .git
           rm -rf .gitignore
           rm -rf .github
+          pip cache purge