Update validate-linux-binaries.yml

pytorch · Sep 23, 2022 · 97034c3 · 97034c3
1 parent decb5fb
commit 97034c3
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/.github/workflows/validate-linux-binaries.yml b/.github/workflows/validate-linux-binaries.yml
@@ -86,7 +86,7 @@ jobs:
     steps:
       - name: Checkout PyTorch builder
         uses: actions/checkout@v2
-      - name: Instlla Conda
+      - name: Install Conda
         uses: conda-incubator/setup-miniconda@v2
         with:
           python-version: ${{ inputs.python_version }}
@@ -102,7 +102,6 @@ jobs:
           DESIRED_CUDA: ${{ matrix.desired_cuda }}
           DESIRED_DEVTOOLSET: ${{ matrix.devtoolset }}
           PACKAGE_TYPE: libtorch
-          LD_LIBRARY_PATH: /usr/local/cuda/lib
         run: |
           sudo apt-get install unzip -y
           set -ex
@@ -111,4 +110,6 @@ jobs:
           conda create -yp ${ENV_NAME} python=${{ inputs.python_version }} numpy
           export LD_LIBRARY_PATH="$(dirname $(which python))/lib"
           export LD_LIBRARY_PATH=${LD_LIBRARY_PATH}:/usr/local/cuda/lib
+          conda run -p ${ENV_NAME} python3  ./test/smoke_test/smoke_test.py
           conda run -p ${ENV_NAME} --cwd libtorch env LD_LIBRARY_PATH=${LD_LIBRARY_PATH} bash ./../check_binary.sh
+          conda env remove -p ${ENV_NAME}