Fix torch version

k2-fsa · Aug 24, 2023 · cbc75ce · cbc75ce
1 parent 443544d
commit cbc75ce
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/.github/scripts/install_torch.sh b/.github/scripts/install_torch.sh
@@ -14,8 +14,9 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-torch=$TORCH_VERSION
-cuda=$CUDA_VERSION
+echo "torch version: $torch"
+echo "cuda version: $cuda"
+
 case ${torch} in
   1.5.*)
     case ${cuda} in
@@ -127,7 +128,7 @@ case ${torch} in
         ;;
     esac
     ;;
-  1.12.1)
+  1.12.*)
     case ${cuda} in
       10.2)
         package="torch==${torch}"