From fffd5af574144af90ec33a6d33f6079b783e0b01 Mon Sep 17 00:00:00 2001
From: Fangjun Kuang <csukuangfj@gmail.com>
Date: Sat, 21 Oct 2023 23:49:42 +0800
Subject: [PATCH] small fixes

---
 scripts/github_actions/build-ubuntu-cuda.sh | 8 +++++---
 scripts/github_actions/install_torch.sh     | 2 ++
 2 files changed, 7 insertions(+), 3 deletions(-)

diff --git a/scripts/github_actions/build-ubuntu-cuda.sh b/scripts/github_actions/build-ubuntu-cuda.sh
index 2d9dab9..f0f3ad2 100755
--- a/scripts/github_actions/build-ubuntu-cuda.sh
+++ b/scripts/github_actions/build-ubuntu-cuda.sh
@@ -18,7 +18,7 @@ fi
 if [ -z $CUDA_VERSION ]; then
   echo "Please set the environment variable CUDA_VERSION"
   echo "Example: export CUDA_VERSION=10.2"
-  # valid values: 10.2, 11.1, 11.3, 11.6, 11.7, 11.8
+  # valid values: 10.2, 11.1, 11.3, 11.6, 11.7, 11.8, 12.1
   exit 1
 fi
 
@@ -43,6 +43,8 @@ make install >/dev/null 2>&1
 
 popd
 
+rm -rf Python-${PYTHON_VERSION}.3
+
 export PATH=$PYTHON_INSTALL_DIR/bin:$PATH
 export LD_LIBRARY_PATH=$PYTHON_INSTALL_DIR/lib:$LD_LIBRARY_PATH
 ls -lh $PYTHON_INSTALL_DIR/lib/
@@ -63,8 +65,8 @@ python3 -m pip install bs4 requests tqdm auditwheel
 echo "Installing torch"
 ./install_torch.sh
 
-rm -rf ~/.cache/pip
-yum clean all
+rm -rf ~/.cache/pip >/dev/null 2>&1
+yum clean all >/dev/null 2>&1
 
 cd /var/www
 
diff --git a/scripts/github_actions/install_torch.sh b/scripts/github_actions/install_torch.sh
index 7ba7485..c63138f 100755
--- a/scripts/github_actions/install_torch.sh
+++ b/scripts/github_actions/install_torch.sh
@@ -13,9 +13,11 @@
 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 # See the License for the specific language governing permissions and
 # limitations under the License.
+set -x
 
 torch=$TORCH_VERSION
 cuda=$CUDA_VERSION
+echo "torch: $torch, cuda: $cuda"
 case ${torch} in
   1.5.*)
     case ${cuda} in