Updated config.yml

facebookresearch · Jul 27, 2022 · 42a7ae9 · 42a7ae9
1 parent 02bb5e9
commit 42a7ae9
Showing 1 changed file with 8 additions and 1 deletion.
diff --git a/.circleci/config.yml b/.circleci/config.yml
@@ -23,6 +23,13 @@ gpu_cu114: &gpu_cu114
     image: ubuntu-2004-cuda-11.4:202110-01
     resource_class: gpu.nvidia.medium
 
+gpu_cu102: &gpu_cu102
+  environment:
+    CUDA_VERSION: "10.2"
+    CUDA_HOME: /usr/local/cuda-10.2
+  machine:
+    image: ubuntu-1604-cuda-10.2:202012-01
+    resource_class: gpu.nvidia.medium
 
 binary_common: &binary_common
   parameters:
@@ -488,4 +495,4 @@ workflows:
             - binary_linux_wheel_py39_cu102
           filters:
             branches:
-              only: binaries_job
+              only: binaries_job