cuda 架构设置
·
import torch
torch.cuda.get_device_capability(0)
添加cmake options：
-DCMAKE_CUDA_ARCHITECTURES=86 -DCMAKE_CUDA_COMPILER=/usr/local/cuda-11.8/bin/nvcc
欢迎来到FlagOS开发社区，这里是一个汇聚了AI开发者、数据科学家、机器学习爱好者以及业界专家的活力平台。我们致力于成为业内领先的Triton技术交流与应用分享的殿堂，为推动人工智能技术的普及与深化应用贡献力量。
更多推荐
3
0- 0
已为社区贡献1条内容
所有评论(0)