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

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

Data Repository
---------------
Gan Quan's avatar
Gan Quan committed
16
* ``DGL_REPO``:
17
    * Values: String (default='https://data.dgl.ai/')
Haibin Lin's avatar
Haibin Lin committed
18
19
    * The repository url to be used for DGL datasets and pre-trained models.
    * Suggested values:
20
21
        * '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
22
23
24
* ``DGL_DOWNLOAD_DIR``:
    * Values: String (default="${HOME}/.dgl")
    * The local directory to cache the downloaded data.
25
26
27
28
29
30
31
32
33
34
35
36

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