install torch

microsoft · Jan 11, 2024 · 927ae2c · 927ae2c
1 parent d9f5927
commit 927ae2c
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/.github/workflows/rllm.yml b/.github/workflows/rllm.yml
@@ -8,6 +8,10 @@ on:
 
 env:
   CARGO_TERM_COLOR: always
+  TORCH_CUDA_ARCH_LIST: 8.0
+  CUDA_COMPUTE_CAP: 80
+  LIBTORCH_USE_PYTORCH: 1
+  LIBTORCH_BYPASS_VERSION_CHECK: 1
 
 jobs:
   build:
@@ -16,6 +20,9 @@ jobs:
 
     steps:
       - uses: actions/checkout@v3
+
+      - run: pip install torch==2.1.0
+
       - uses: Jimver/[email protected]
         id: cuda-toolkit
         with: