env_var.rst 900 Bytes
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.