From d4a1be3266857c963e4a8e76746baeba3be8f369 Mon Sep 17 00:00:00 2001
From: Quan Gan <coin2028@hotmail.com>
Date: Wed, 7 Apr 2021 04:41:15 +0000
Subject: [PATCH] update meta

---
 conda/dgl/meta.yaml | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/conda/dgl/meta.yaml b/conda/dgl/meta.yaml
index 33a1acbea4c9..2d953cddb722 100644
--- 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