env_var.rst 1.35 KB
Newer Older
Haibin Lin's avatar
Haibin Lin committed
1
2
3
Environment Variables
=====================

4
5
6
7
8
9
Global Configurations
---------------------
* ``DGLDEFAULTDIR``:
    * Values: String (default=``"${HOME}/.dgl"``)
    * The directory to save the DGL configuration files.

Haibin Lin's avatar
Haibin Lin committed
10
11
Backend Options
---------------
Gan Quan's avatar
Gan Quan committed
12
* ``DGLBACKEND``:
Haibin Lin's avatar
Haibin Lin committed
13
    * Values: String (default='pytorch')
14
    * The backend deep learning framework for DGL.
Haibin Lin's avatar
Haibin Lin committed
15
    * Choices:
Jinjing Zhou's avatar
Jinjing Zhou committed
16
17
        * 'pytorch': use PyTorch as the backend implementation.        
        * 'tensorflow': use Apache TensorFlow as the backend implementation.
Haibin Lin's avatar
Haibin Lin committed
18
19
20
21
        * 'mxnet': use Apache MXNet as the backend implementation.

Data Repository
---------------
Gan Quan's avatar
Gan Quan committed
22
* ``DGL_REPO``:
23
    * Values: String (default='https://data.dgl.ai/')
Haibin Lin's avatar
Haibin Lin committed
24
25
    * The repository url to be used for DGL datasets and pre-trained models.
    * Suggested values:
26
27
        * 'https://data.dgl.ai/': DGL repo for Global Region.
        * 'https://dgl-data.s3.cn-north-1.amazonaws.com.cn/': DGL repo for Mainland China
Gan Quan's avatar
Gan Quan committed
28
* ``DGL_DOWNLOAD_DIR``:
29
    * Values: String (default=``"${HOME}/.dgl"``)
Gan Quan's avatar
Gan Quan committed
30
    * The local directory to cache the downloaded data.
31
32
33
34
35
36
37
38
39
40
41
42

Intel CPU Performance Options
---------------
* ``DGL_CPU_INTEL_KERNEL_ENABLED``:
    * Values: int (default='0')
    * Use dynamic cpu kernels.
    * Suggested values: 1

* ``DGL_CPU_INTEL_KERNEL_LOG``:
    * Values: int (default='0')
    * Show diagnostic message (debug mode).
    * Suggested values: 1