added Cudnn roles

2021-05-25 23:10:22 +04:00
parent c62e0ca288
commit 7742377304
25 changed files with 168 additions and 66 deletions
--- a/sample-run/tensorflow/packages.yml
+++ b/sample-run/tensorflow/packages.yml
@@ -26,8 +26,11 @@
    #   vars:
    #     users: "{{ secrets.users }}"

-    - role: cuda # CUDA Driver Setup
-      tags: [cuda]
+#    - role: cuda # CUDA Driver Setup
+#      tags: [cuda]
+
+    - role: cudnn
+      tags: [cudnn]

    # - role: nvidia # NVIDIA-Docker Runtime Setup
    #   tags: [nvidia]