singa-dev mailing list archives

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

chrishkchris commented 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 
   
   Then, 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