我发现了一个关于错误的错误:无法在tensorflow CUDA_DIR api中找到libdevice目录${ object_detection }/nvvm/libdevice
我发现了一个关于错误的错误:无法在tensorflow CUDA_DIR api中找到libdevice目录${ object_detection }/nvvm/libdevice
提问于 2022-06-04 11:36:47
Windows版本: Windows 10 Pro 21H2 19044.1706 GPU: rtx2070
import tensorflow as tf
import torch
print(torch.__version__) #1.10.1+cu113
print(torch.version.cuda) #11.3
print(tf.__version__) #2.9.1我就跑了
python .\object_detection\builders\model_builder_tf2_test.py我可以得到‘运行24测试在18.279 s OK (skipped=1)’的结果；
但是当我想训练我的模特时，我用
feature_extractor {
type: 'faster_rcnn_inception_resnet_v2_keras'
}在我的pipeline_config中，我运行
python .\object_detection\model_main_tf2.py --logtostderr --pipeline_config_path=LOCATION_OF_MY_PIPECONFIG --model_dir=LOCATION_OF_MY_MODEL_DIR然后我得到以下错误
在我的系统环境变量中，'CUDA_DIR‘是可变的，可以访问
回答 2
Stack Overflow用户
发布于 2022-06-15 00:15:29
我也有同样的问题，只是把它修好了。即使设置了"CUDA_DIR“，库也找不到文件夹，因为它没有使用该变量或我尝试过的任何其他变量。这个帖子有助于理解这个问题。我能找到的唯一解决方案就是复制所需的文件。
快速修复的步骤：
- 查找您的CUDA nvvm安装位置(对我来说，它是"C:\Program \NVIDIA GPU计算工具包\CUDA\v11.6“)。
- 查找脚本的工作目录(环境或运行脚本的目录)。
- 将整个nvvm文件夹复制到工作目录中，您的脚本就可以工作了。
这不是一个很好的解决方案，但在别人发布答案之前，您至少可以运行您的代码。
Stack Overflow用户
发布于 2022-09-27 14:23:19
在虚拟环境存在的目录中复制CUDA nvvm。
页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持
原文链接：
https://stackoverflow.com/questions/72499414
复制相关文章
相似问题
腾讯云开发者
