小编Shu*_*Fan的帖子

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

我可以获得“在 18.279 秒内进行 24 次测试，正常（跳过 = 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”是变量并且可以访问