How to build with CUDA11+?

ethnhe / PVN3D

Code for "PVN3D: A Deep Point-wise 3D Keypoints Hough Voting Network for 6DoF Pose Estimation", CVPR 2020

MIT License

493 stars 105 forks source link

Closed Ultraopxt closed 3 years ago

Ultraopxt commented 3 years ago

May I ask How to build with CUDA11+? Cause my GPU is 30 series

LEONHWH commented 2 years ago

请问如何用CUDA11+构建？因为我的 GPU 是 30 系列

你好，你好吗？你用30系列的显卡能通过成功实现“python3 setup.py build_ext”的步骤 CUDA11+以上需要配置pytorch==1.7以上，我想是否能通过部分代码修改或者更换一些配置版本来成功跑通代码？期待您的回复，谢谢！

HannahHaensen commented 2 years ago

@LEONHWH or @qingweihk did you find a solution to compile it with Cuda 11 and pytorch 1.8 or higher?