jeya-maria-jose / KiU-Net-pytorch

Official Pytorch Code of KiU-Net for Image/3D Segmentation - MICCAI 2020 (Oral), IEEE TMI
https://sites.google.com/view/kiunet/kiu-net
MIT License
357 stars 80 forks source link

About the environment #39

Open zhouyizhuo opened 10 months ago

zhouyizhuo commented 10 months ago

I am having the following problem,can someone give some help?thanks! image

Here is my CUDA version and cuDNN version: image

zhouyizhuo commented 10 months ago

And which torch version should I choose?1.0.1.post2 or 1.4 ? image image