Skip to content

GitLab

Unverified Commit 70ee8664 authored by Da Zheng's avatar Da Zheng Committed by GitHub
Browse files

[Docker]Move to the latest MXNet version. (#610) 

* use the right version of mxnet.

* fix.
parent 653428bd
Show whitespace changes
Inline Side-by-side
Showing with 3 additions and 3 deletions
docker/install/ubuntu_install_mxnet_cpu.sh
View file @ 70ee8664
pip3 install mxnet==1.5.0b20190523 pip3 install mxnet==1.5.0b20190605
docker/install/ubuntu_install_mxnet_gpu.sh
View file @ 70ee8664
pip3 install mxnet-cu90==1.5.0b20190523 pip3 install mxnet-cu90==1.5.0b20190605
python/dgl/backend/mxnet/tensor.py
View file @ 70ee8664
...@@ -13,7 +13,7 @@ MX_VERSION = LooseVersion(mx.__version__) ...@@ -13,7 +13,7 @@ MX_VERSION = LooseVersion(mx.__version__)
# After MXNet 1.5, empty tensors aren't supprted by default. # After MXNet 1.5, empty tensors aren't supprted by default.
# after we turn on the numpy compatible flag, MXNet supports empty NDArray. # after we turn on the numpy compatible flag, MXNet supports empty NDArray.
if MX_VERSION.version[0] == 1 and MX_VERSION.version[1] >= 5: 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(): def data_type_dict():
return {'float16' : np.float16, 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