问如何使用指定的cuda和cudnn构建caffe2？
EN

Stack Overflow用户

提问于 2018-07-18 23:45:42

回答 1查看 207关注 0票数 0

在服务器端，cudnn 7.0和cuda 9.1位于/usr/local/cuda。我需要使用detectron，但它不支持cuda 9.1，所以我需要安装cuda 8.0。

我尝试安装cuda8.0和cudnn，第二个答案是在/ this /xx/caffe2。

然后构建caffe2，但它再次使用/usr/local/cuda处的cuda。

那么有没有办法在/home/xx/caffe2中使用cuda呢？

发布于 2018-09-12 02:38:43

您需要在运行安装时指定环境变量：

您可以在存储库中的setup.py文件中找到环境变量的完整描述。

例如：

CUDA_HOME=/home/xxx/cuda-9.1 python setup.py install

票数 0

页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持

原文链接：

https://stackoverflow.com/questions/51406124

复制

相似问题

问如何使用指定的cuda和cudnn构建caffe2？EN