SINGA-429 update dockerfile for cuda cudnn update #437
Conversation
|
Also, please kindly advise do we need cudnn 7.5? as mentioned in https://issues.apache.org/jira/browse/SINGA-429 However, the official docker image from nvidia is offering cudnn 7.4.2 |
dcslin
changed the title
|
pls add the command for compiling and installing singa. |
|
Hi @nudles , this is almost done. I have been stuck on a weird cmake(version > 3.12) behaviour and investigating it: |
|
how about mkldnn? can install it in the docker and let singa use it? |
|
Hi @nudles , I am working on MKL, but I am trying make it optional, since for |
|
MKLDNN is called when the code is running on CPU; cudnn is called when it is running on GPU. |
|
HI @nudles , this is basically done:
but it is not yet ready to merge and depends on:
|
| # ENV CMAKE_INCLUDE_PATH /usr/local/cuda/include:${CMAKE_INCLUDE_PATH} | ||
| # ENV CMAKE_LIBRARY_PATH /usr/local/cuda/lib64:${CMAKE_LIBRARY_PATH} | ||
| # config ssh service | ||
| && mkdir /var/run/sshd \ | ||
| # install mkldnn | ||
| RUN git clone https://github.com/intel/mkl-dnn.git /tmp/mkl-dnn \ |
There was a problem hiding this comment.
it would be better to fix the version, e.g., from https://github.com/intel/mkl-dnn/archive/v1.0-pc.zip
otherwise, when the github repo is updated with API changes; the compilation of singa may fail.
There was a problem hiding this comment.
thank you, this is right, I have fixed it.
| libgoogle-glog-dev \ | ||
| sudo \ | ||
| && apt-get clean \ | ||
| && apt-get autoremove \ |
There was a problem hiding this comment.
do you need to install icc and icpc to compile mkldnn?
There was a problem hiding this comment.
hi i will review this. for now with "manual fixing cmake file" i can execute mkldnn test
Hi team,
Kindly find the changes below regarding update of cuda and cudnn version.
Kindly let me know your comment. Thank you