我收到一个关于错误的错误：在tensorflow object_detection api中找不到libdevice目录${CUDA_DIR}/nvvm/libdevice。

7 浏览
0 Comments

我收到一个关于错误的错误：在tensorflow object_detection api中找不到libdevice目录${CUDA_DIR}/nvvm/libdevice。

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

我可以得到结果'Ran 24 tests in 18.279s OK (skipped=1)'；

但是当我想要训练我的模型时，我在我的pipeline_config中使用了

feature_extractor {
   type: 'faster_rcnn_inception_resnet_v2_keras'
}

然后我运行

python .\object_detection\model_main_tf2.py --logtostderr --pipeline_config_path=LOCATION_OF_MY_PIPECONFIG --model_dir=LOCATION_OF_MY_MODEL_DIR

然后我得到以下错误

enter image description here

在我的系统环境变量中，'CUDA_DIR'是一个可访问的变量。

0