Closed KeunwooPark closed 5 years ago
I found out what was wrong. I was using GTX 1060 and it had Pascal Architecture. So I had to use cudnn 5, not 4. Even though the Dockerfile for gpu uses cudnn5, it does not use Caffe SegNet repo for cudnn 5. The solution is to change the git repo in the Docker file to caffe-segnet-cudnn5. Moreover, I had to use cuda 8.0 and Ubuntu 16.04 for the Docker environment. I think it is because of the GPU architecture issue.
I tried to build the gpu docker image on Ubuntu 16.04. Dockerfile fails with the below error.