CannyLab/tsne-cuda

CUDA error 209 no kernel image is available for execution on the device

Closed this issue · 6 comments

JXuann commented

Hi,
I just installed tsnecuda via conda (on Ubuntu 20).

When running tsnecuda.test(), I encountered this error:

Initializing cuda handles... done.
KNN Computation... Faiss assertion 'err__ == cudaSuccess' failed in void faiss::gpu::runL2Norm(faiss::gpu::Tensor<T, 2, true, IndexType>&, bool, faiss::gpu::Tensor<float, 1, true, IndexType>&, bool, cudaStream_t) [with T = float; TVec = float4; IndexType = int; cudaStream_t = CUstream_st*] at /home/conda/feedstock_root/build_artifacts/faiss-split_1636459943780/work/faiss/gpu/impl/L2Norm.cu:323; details: CUDA error 209 no kernel image is available for execution on the device
Aborted (core dumped)

I looked at the solved old issues and realized it could be compute compatibility. I probably need to define --gencode=arch=compute_50,code=sm_50 (for quadro m series) somewhere. But I do not know where could I specify it or how I should solve this.

Could you please give me some suggestions? Many thanks!

DavidMChan commented

So this is actually an issue with the linked version of FAISS, since we already compile using the 5.0 compute capability (See:

tsne-cuda/CMakeLists.txt

Line 66 in e565c6b

-gencode=arch=compute_50,code=sm_50
). The way to get around this would be by first building FAISS's GPU code with the correct compute capability (which should be possible on CUDA 11), and then going back and linking tsnecuda against the generated FAISS code. You'll basically have to follow the instructions in the docker file to build FAISS, but alter the line here: https://github.com/facebookresearch/faiss/blob/79e74fe3075e494abcc909b4988ef3c3cb059f72/.circleci/Dockerfile.faiss_gpu#L25
to support the correct compute architecture.

DavidMChan commented

I'm going to close this issue, since it's a problem upstream, but I'm happy to answer questions in the replies.

JXuann commented

Hi @DavidMChan ,

Thanks for your quick reply! Much appreciated.

How do I linking tsnecuda against the generated FAISS code?

DavidMChan commented

This should happen automatically when building tsnecuda from scratch with FAISS installed (since this code will pick up the location of the FAISS shared library:

tsne-cuda/CMakeLists.txt

Line 154 in e565c6b

find_package(FAISS REQUIRED)

If it's not found, you can specify the location of the faiss libraries by setting the CMAKE variables that are set in this script: https://github.com/CannyLab/tsne-cuda/blob/main/cmake/Modules/FindFAISS.cmake

JXuann commented

Ok, thanks. Just to confirm: I need to completely remove the existing tsnecuda (previously installed via conda), build FAISS-gpu and then build tsnecuda from source (as listed here: https://github.com/CannyLab/tsne-cuda/blob/master/INSTALL.md)?

DavidMChan commented