From 92a68086173dd549afba0415ffa5d2675d617234 Mon Sep 17 00:00:00 2001 From: anjith2006 <ageorge@idiap.ch> Date: Thu, 13 Feb 2020 15:08:45 +0100 Subject: [PATCH] cuda --- conda/meta.yaml | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/conda/meta.yaml b/conda/meta.yaml index b57a64b..034b940 100644 --- a/conda/meta.yaml +++ b/conda/meta.yaml @@ -38,7 +38,7 @@ requirements: - docopt {{ docopt }} - matplotlib {{ matplotlib }} - h5py {{ h5py }} - - cudatoolkit {{ cudatoolkit }} # [linux] + - cudatoolkit==9.0 # [linux] - tensorflow {{ tensorflow }} - pytorch {{ pytorch }} # [linux] - torchvision {{ torchvision }} # [linux] -- GitLab