diff --git a/.github/workflows/run-docker-image.yml b/.github/workflows/run-docker-image.yml
index ccc11d06f..0011d6cca 100644
--- a/.github/workflows/run-docker-image.yml
+++ b/.github/workflows/run-docker-image.yml
@@ -32,6 +32,8 @@ jobs:
 
               nvcc --version
 
+              ls -lh /usr/local/cuda-*/compat/*
+
               # For torch1.9.0-cuda10.2
               export LD_LIBRARY_PATH=/usr/local/cuda-10.2/compat:$LD_LIBRARY_PATH
 
@@ -41,6 +43,12 @@ jobs:
               # For torch2.0.0-cuda11.7
               export LD_LIBRARY_PATH=/usr/local/cuda-11.7/compat:$LD_LIBRARY_PATH
 
+              # For torch2.1.0-cuda11.8
+              export LD_LIBRARY_PATH=/usr/local/cuda-11.8/compat:$LD_LIBRARY_PATH
+
+              # For torch2.1.0-cuda12.1
+              export LD_LIBRARY_PATH=/usr/local/cuda-12.1/compat:$LD_LIBRARY_PATH
+
 
               which nvcc
               cuda_dir=$(dirname $(which nvcc))