Unet-Segmentation icon indicating copy to clipboard operation
Unet-Segmentation copied to clipboard

Published 20 hours ago • verified publisher icon lmb-freiburg

CUDA 10.1 support

Open Hir320 opened this issue 3 years ago • 3 comments

Hello all,

Does U-Net run with CUDA 10.1? I have used U-Net with CUDA 10.0, but currently I want to build U-Net on CUDA 10.1 for some reasons. My server's OS is Ubuntu 18.04.

Do I just need to follow "Setup on own server (from source)" to build it on CUDA 10.1? Does it require cuDNN before installation of U-Net?

Hir320 avatar Nov 03 '21 03:11 Hir320

Finally, my server got recovered, so it is now compatible with CUDA 10.0. Thus I don't need to build U-Net on 10.1. However, I still wonder if U-Net is compatible with CUDA 10.1, so I leave this issue open.

Thank you

Hir320 avatar Nov 23 '21 17:11 Hir320

Building from source might be possible. In the worst case there are interface changes in cuDNN, so that cuDNN function calls must be slightly adapted. Please check the original caffe sources for updates in question.

ThorstenFalk avatar Dec 01 '21 20:12 ThorstenFalk

@ThorstenFalk Thank you. I will try and figure out if problems happen.

Hir320 avatar Dec 28 '21 02:12 Hir320