$ nix run "(with import <nixpkgs> { config = { allowUnfree=true; cudaSupport=true; cudnnSupport=true; }; };
python3.withPackages (ps: [ ps.mxnet ]))"

builder for '/nix/store/f5zwrjv4ad0y8p2qk6472d4ryx7nvzm8-mxnet-1.2.1.drv' failed with exit code 1; last 10 log lines:
  CMake Error: CMAKE_CUDA_COMPILER not set, after EnableLanguage
  CMake Error at cmake/FirstClassLangCuda.cmake:79 (try_run):
    Failed to configure test project build system.
  Call Stack (most recent call first):
    cmake/FirstClassLangCuda.cmake:147 (mshadow_detect_installed_gpus)
    CMakeLists.txt:534 (mshadow_select_nvcc_arch_flags)
  
  -- Configuring incomplete, errors occurred!
  See also "/build/apache-mxnet-src-1.2.1-incubating/build/CMakeFiles/CMakeOutput.log".
cannot build derivation '/nix/store/cqkmsv1kldxahdl6klzgn42p6s0vwwq3-python3.6-mxnet-1.2.1.drv': 1 dependencies couldn't be built
cannot build derivation '/nix/store/xxha232lplcg7zzgv52lmd8qh43bpwqn-python3-3.6.6-env.drv': 1 dependencies couldn't be built
[0 built (1 failed), 0.0 MiB DL]
error: build of '/nix/store/xxha232lplcg7zzgv52lmd8qh43bpwqn-python3-3.6.6-env.drv' failed