Sign in
apache
/
mxnet-test
/
refs/tags/v0.10.11
/
.
/
example
/
image-classification
/
common
/
find_mxnet.py
blob: a24444306721814c1845350306ef8fa1754e607d [
file
] [
log
] [
blame
]
import
os
,
sys
os
.
environ
[
"MXNET_CUDNN_AUTOTUNE_DEFAULT"
]
=
"1"
try
:
import
mxnet
as
mx
except
ImportError
:
curr_path
=
os
.
path
.
abspath
(
os
.
path
.
dirname
(
__file__
))
sys
.
path
.
append
(
os
.
path
.
join
(
curr_path
,
"../../../python"
))
import
mxnet
as
mx