Tesla P100的arch填坑!!!

pytorch-faster-rcnn中的make.sh脚本中的关于GPU model对应的arch,这个真的很坑,由于本人的服务器是Tesla P100的,但是呢,一直因为是Tesla k80!!!
根据这篇博客:https://blog.csdn.net/kkk584520/article/details/53814067
了解到Tesla P100对应arch是sm_60,因为只需要在make.sh脚本中选择sm_60的那行就行了~

猜你喜欢

转载自blog.csdn.net/qq_18649781/article/details/89849876