update meta

dmlc · Apr 7, 2021 · d4a1be3 · d4a1be3
1 parent 1538229
commit d4a1be3
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/conda/dgl/meta.yaml b/conda/dgl/meta.yaml
@@ -1,3 +1,5 @@
+{% set use_cuda = environ.get('USE_CUDA', False) %}
+{% set package_name = 'dgl' + environ.get('DGL_PACKAGE_SUFFIX', '') %}
 package:
   name: dgl{{ environ.get('DGL_PACKAGE_SUFFIX', '') }}
   version: "0.6.1"
@@ -19,10 +21,13 @@ requirements:
     - scipy
     - networkx
     - requests
+    - vs2015_runtime # [win]
+    - cudatoolkit {{ CUDA_VER }} # [use_cuda]
 
 build:
   script_env:
     - USE_CUDA
+    - CUDA_VER
 
 about:
   home: https://github.com/dmlc/dgl.git