singa-dev mailing list archives

Site index · List index
Message view « Date » · « Thread »
Top « Date » · « Thread »
From GitBox <...@apache.org>
Subject [GitHub] [singa] chrishkchris edited a comment on pull request #709: Upgrade CUDNN version from 7.3.1 to 7.6.5
Date Fri, 29 May 2020 12:38:30 GMT

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:
users@infra.apache.org



Mime
View raw message