This repository was archived by the owner on Nov 17, 2023. It is now read-only.

Description
Description
When I install mxnet from source, it gives me nvcc fatal unsupported gpu architecture 'compute_86'. My cuda version is 11.0. I can alleviate the issue by change Auto to ;8.0.
https://github.com/apache/incubator-mxnet/blob/b54b7f36c944b933f2aa94f224821548da5fcaf6/CMakeLists.txt#L43
I think the current mxnet installation may not handle cuda 11.0 well.
To Reproduce
rm -rf build
mkdir -p build && cd build
cmake -GNinja \
-DUSE_CUDA=ON \
-DUSE_MKL_IF_AVAILABLE=ON \
-DCMAKE_CUDA_COMPILER_LAUNCHER=ccache \
-DCMAKE_C_COMPILER_LAUNCHER=ccache \
-DCMAKE_CXX_COMPILER_LAUNCHER=ccache \
-DCMAKE_BUILD_TYPE=Release \
..
ninja