chrishkchris edited a comment on pull request #709: URL: https://github.com/apache/singa/pull/709#issuecomment-635948985
I found the problem: in nvidia/cuda:10.2-devel-ubuntu16.04, the location of cublas header moved to another directories, To simply solve it, I provided a new docker container of cuda10.2 specifically for the conda build using cuda10.2. at tool/conda/docker/cuda10.2/Dockerfile Also, in order to upgrade to cuda 10.2 the nccl needed to be upgraded to 2.6.4. But I keep locking nccl 2.4.8 using cuda 10.0 for performance reason. I think PR is ready for merge. ---------------------------------------------------------------- This is an automated message from the Apache Git Service. To respond to the message, please log on to GitHub and use the URL above to go to the specific comment. For queries about this service, please contact Infrastructure at: [email protected]
