pycuda库pip安装失败解决

在使用tensorrt调用已经生成好的.engine模型，但是需要pycuda库，pip安装屡屡失败。后来发现是python版本不对，之前conda虚拟环境都是python=3.8。尝试过类似方法都不行，换numpy，settools版本也失败。

2301_80667554

528人浏览 · 2024-07-30 14:06:46

2301_80667554 · 2024-07-30 14:06:46 发布

在使用tensorrt调用已经生成好的.engine模型，但是需要pycuda库，pip安装屡屡失败

尝试过类似方法都不行，换numpy，settools版本也失败

解决 pip install 出现 error: subprocess-exited-with-error 错误的方法_pip install error: subprocess-exited-with-error-CSDN博客

后来发现是python版本不对，之前conda虚拟环境都是python=3.8

conda create -n [name] python=3.9
pip install pycuda

这样就行了

Triton中文社区

欢迎来到由智源人工智能研究院发起的Triton中文社区，这里是一个汇聚了AI开发者、数据科学家、机器学习爱好者以及业界专家的活力平台。我们致力于成为业内领先的Triton技术交流与应用分享的殿堂，为推动人工智能技术的普及与深化应用贡献力量。

更多推荐

torch.device(‘cuda:0‘ if cuda else ‘cpu‘)

# 这个地方的cuda:0 实际上并不是0号GPU，他取决于CUDA_VISIBLE_DEVICES# 然后逻辑GPU和物理GPU有一个对应关系# 如果CUDA_VISIBLE_DEVICES为2,1,3# 那么CUDA:0就是2号GPU， CUDA:1 就是1号GPU CUDA:3 就是3号GPUreturn torch.device('cuda:0' if cuda else 'cpu')..