From 3d5d60efab3cfc66fee66beb216ac164e8fe72cf Mon Sep 17 00:00:00 2001 From: Fangjun Kuang Date: Sat, 1 Feb 2025 20:17:28 +0800 Subject: [PATCH] fix python --- scripts/github_actions/build-ubuntu-cuda.sh | 45 +++++++++++---------- 1 file changed, 23 insertions(+), 22 deletions(-) diff --git a/scripts/github_actions/build-ubuntu-cuda.sh b/scripts/github_actions/build-ubuntu-cuda.sh index f25d9b4..d312f03 100755 --- a/scripts/github_actions/build-ubuntu-cuda.sh +++ b/scripts/github_actions/build-ubuntu-cuda.sh @@ -27,6 +27,29 @@ if [[ $TORCH_VERSION =~ 2.2.* && $CUDA_VERSION =~ 12.* ]]; then export TORCH_CUDA_ARCH_LIST="8.0 8.6 8.9 9.0" fi +# python3 -m pip install scikit-build +python3 -m pip install -U pip cmake +python3 -m pip install wheel twine typing_extensions +python3 -m pip install bs4 requests tqdm auditwheel + +echo "Installing torch" +./install_torch.sh + +python3 -c "import torch; print(torch.__file__)" + + +# -- Autodetected CUDA architecture(s): 5.0;8.0;8.6;8.9;9.0;9.0a +# CMake Error at /Python-3.8.2/py-3.8/lib/python3.8/site-packages/torch/share/cmake/Caffe2/Modules_CUDA_fix/upstream/FindCUDA/select_compute_arch.cmake:227 (message): +# Unknown CUDA Architecture Name 9.0a in CUDA_SELECT_NVCC_ARCH_FLAGS +# Call Stack (most recent call first): +# /Python-3.8.2/py-3.8/lib/python3.8/site-packages/torch/share/cmake/Caffe2/public/utils.cmake:401 (cuda_select_nvcc_arch_flags) +# /Python-3.8.2/py-3.8/lib/python3.8/site-packages/torch/share/cmake/Caffe2/public/cuda.cmake:342 (torch_cuda_get_nvcc_gencode_flag) +# /Python-3.8.2/py-3.8/lib/python3.8/site-packages/torch/share/cmake/Caffe2/Caffe2Config.cmake:87 (include) +# /Python-3.8.2/py-3.8/lib/python3.8/site-packages/torch/share/cmake/Torch/TorchConfig.cmake:68 (find_package) +# cmake/torch.cmake:14 (find_package) +# CMakeLists.txt:62 (include) +sed -i.bak /9.0a/d /Python-*/py-3.*/lib/python3.*/site-packages/torch/share/cmake/Caffe2/Modules_CUDA_fix/upstream/FindCUDA/select_compute_arch.cmake || true + if [[ x"$IS_2_28" != x"1" ]]; then yum -y install openssl-devel fi @@ -62,28 +85,6 @@ ls -lh $PYTHON_INSTALL_DIR/lib/ python3 --version which python3 -# python3 -m pip install scikit-build -python3 -m pip install -U pip cmake -python3 -m pip install wheel twine typing_extensions -python3 -m pip install bs4 requests tqdm auditwheel - -echo "Installing torch" -./install_torch.sh - -python3 -c "import torch; print(torch.__file__)" - -# -- Autodetected CUDA architecture(s): 5.0;8.0;8.6;8.9;9.0;9.0a -# CMake Error at /Python-3.8.2/py-3.8/lib/python3.8/site-packages/torch/share/cmake/Caffe2/Modules_CUDA_fix/upstream/FindCUDA/select_compute_arch.cmake:227 (message): -# Unknown CUDA Architecture Name 9.0a in CUDA_SELECT_NVCC_ARCH_FLAGS -# Call Stack (most recent call first): -# /Python-3.8.2/py-3.8/lib/python3.8/site-packages/torch/share/cmake/Caffe2/public/utils.cmake:401 (cuda_select_nvcc_arch_flags) -# /Python-3.8.2/py-3.8/lib/python3.8/site-packages/torch/share/cmake/Caffe2/public/cuda.cmake:342 (torch_cuda_get_nvcc_gencode_flag) -# /Python-3.8.2/py-3.8/lib/python3.8/site-packages/torch/share/cmake/Caffe2/Caffe2Config.cmake:87 (include) -# /Python-3.8.2/py-3.8/lib/python3.8/site-packages/torch/share/cmake/Torch/TorchConfig.cmake:68 (find_package) -# cmake/torch.cmake:14 (find_package) -# CMakeLists.txt:62 (include) -sed -i.bak /9.0a/d /Python-*/py-3.*/lib/python3.*/site-packages/torch/share/cmake/Caffe2/Modules_CUDA_fix/upstream/FindCUDA/select_compute_arch.cmake || true - rm -rf ~/.cache/pip >/dev/null 2>&1 yum clean all >/dev/null 2>&1