NiftyNet培训兼容的GPU卡

我目前有GeForce RTX 2060 w / 6GB vram,技嘉Aourus,并且在使用niftynet时遇到了问题。我已经尝试过通过降级到tensorflow-gpu 1.13.2和numpy 1.16的各种建议,将cudnn更改为7.4,并尝试将util_common.py更改为包括内存增长控制     #config.gpu_options.allow_growth =真     #config.gpu_options.per_process_gpu_memory_fraction = 0.5 但是我仍然继续使GPU-> CPU memcpy失败,并且无法创建cudnn句柄:CUDNN_STATUS_INTERNAL_ERROR

我开始怀疑该显卡是否不兼容? niftynet已成功部署到哪些图形卡?你有什么经验?

tsm1010 回答:NiftyNet培训兼容的GPU卡

暂时没有好的解决方案,如果你有好的解决方案,请发邮件至:iooj@foxmail.com
本文链接:https://www.f2er.com/2955354.html

大家都在问