From 6eb8945da19c28876b311adf41a6ce462107fda1 Mon Sep 17 00:00:00 2001
From: Amir MOHAMMADI <amir.mohammadi@idiap.ch>
Date: Mon, 18 Oct 2021 16:28:45 +0200
Subject: [PATCH] Use pytorch 1.8.0 cuda version

---
 bob/devtools/data/conda_build_config.yaml | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/bob/devtools/data/conda_build_config.yaml b/bob/devtools/data/conda_build_config.yaml
index fa337864..3ee0130f 100644
--- a/bob/devtools/data/conda_build_config.yaml
+++ b/bob/devtools/data/conda_build_config.yaml
@@ -426,7 +426,7 @@ pytest_cov:
 python_graphviz:
   - 0.17
 pytorch:
-  - 1.8.0
+  - 1.8.0 cuda*
 pyyaml:
   - 5.4.1
 pyzmq:
-- 
GitLab