search

FederatedAI / Practicing-Federated-Learning

Apache License 2.0
795 stars 245 forks source link

第十章的CUPY以及运行问题 #15

Open 876720687 opened 1 year ago

876720687 commented 1 year ago

想问一下这个代码的运行环境是什么？cupy-cuda11x好像没有支持到？ model/utils/nms/non_maximum_suppression.py文件中的@cp.util.memoize()以及下面的函数在Cuda11.6当中没有支持版本。