From 2569bf66b6b3f9b530abaf24db3a7a068a381d00 Mon Sep 17 00:00:00 2001 From: Fangjun Kuang Date: Sat, 1 Feb 2025 21:01:56 +0800 Subject: [PATCH] fix importing torch --- .github/workflows/ubuntu-cuda-wheels.yml | 5 +++++ 1 file changed, 5 insertions(+) diff --git a/.github/workflows/ubuntu-cuda-wheels.yml b/.github/workflows/ubuntu-cuda-wheels.yml index 82aea4d..219968e 100644 --- a/.github/workflows/ubuntu-cuda-wheels.yml +++ b/.github/workflows/ubuntu-cuda-wheels.yml @@ -108,14 +108,19 @@ jobs: python_dir=/opt/python/cp38-cp38 elif [[ $PYTHON_VERSION == "3.9" ]]; then python_dir=/opt/python/cp39-cp39 + export PYTHONPATH=/opt/python/cp39-cp39/lib/python3.9/site-packages:$PYTHONPATH elif [[ $PYTHON_VERSION == "3.10" ]]; then python_dir=/opt/python/cp310-cp310 + export PYTHONPATH=/opt/python/cp310-cp310/lib/python3.10/site-packages:$PYTHONPATH elif [[ $PYTHON_VERSION == "3.11" ]]; then python_dir=/opt/python/cp311-cp311 + export PYTHONPATH=/opt/python/cp311-cp311/lib/python3.11/site-packages:$PYTHONPATH elif [[ $PYTHON_VERSION == "3.12" ]]; then python_dir=/opt/python/cp312-cp312 + export PYTHONPATH=/opt/python/cp312-cp312/lib/python3.12/site-packages:$PYTHONPATH elif [[ $PYTHON_VERSION == "3.13" ]]; then python_dir=/opt/python/cp313-cp313 + export PYTHONPATH=/opt/python/cp313-cp313/lib/python3.13/site-packages:$PYTHONPATH else echo "Unsupported Python version $PYTHON_VERSION" exit 1