Skip to content
Snippets Groups Projects
Commit 77f70f8e authored by calrama's avatar calrama
Browse files

centos-7: mxnet: Limit cuda architectures to 6.1 (bin) and 3.0, 5.2, 7.5 (ptx)

parent 9ad8fa48
No related branches found
No related tags found
No related merge requests found
......@@ -205,6 +205,9 @@ RUN version=1.5.1 && \
-D CMAKE_CXX_STANDARD=14 -D USE_CXX14_IF_AVAILABLE=ON \
-D CMAKE_BUILD_TYPE=Release -D CMAKE_INSTALL_PREFIX=/usr/local \
-D USE_CUDA=ON -D USE_CUDNN=ON -D USE_NCCL=OFF \
-D CUDA_ARCH_NAME=Manual \
-D CUDA_ARCH_BIN="61" \
-D CUDA_ARCH_PTX="30 52 75" \
-D USE_OPENCV=ON -D USE_OPENMP=ON \
-D USE_LAPACK=ON -D USE_MKL_IF_AVAILABLE=OFF \
-D USE_PROFILER=ON \
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment