Skip to content

GitLab

Commit ada3786c authored by Lingfan Yu's avatar Lingfan Yu Committed by Da Zheng
Browse files

[Docker]Update MXNet CI Image (#582) 

* use mxnet 20190528 build

* fix mxnet api name
parent bd2d2c75
Hide whitespace changes
Inline Side-by-side
Showing with 3 additions and 3 deletions
docker/install/ubuntu_install_mxnet_cpu.sh
View file @ ada3786c
pip3 install mxnet==1.4.1
pip3 install mxnet==1.5.0b20190528
docker/install/ubuntu_install_mxnet_gpu.sh
View file @ ada3786c
pip3 install mxnet-cu90==1.4.1
pip3 install mxnet-cu90==1.5.0b20190528
python/dgl/backend/mxnet/tensor.py
View file @ ada3786c
......@@ -11,7 +11,7 @@ MX_VERSION = LooseVersion(mx.__version__)
# After MXNet 1.5, empty tensors aren't supprted by default.
# after we turn on the numpy compatible flag, MXNet supports empty NDArray.
if MX_VERSION.version[0] == 1 and MX_VERSION.version[1] >= 5:
mx.set_np_compat(True)
mx.set_np_shape(True)
def data_type_dict():
return {'float16' : np.float16,
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment