Не могу определить устройство cuda - PullRequest
0 голосов
/ 04 апреля 2020

Python 3.6.9

Поддержка Tensorflow 2.1.0 GPU

Cuda 10.0

TensorRT 6.0

GPU - 1

Используется для преобразования сохраненной модели Tensorflow в TensorRT (https://docs.nvidia.com/deeplearning/frameworks/tf-trt-user-guide/index.html)

Я получаю ошибку даже после export CUDA_VISIBLE_DEVICES=0 или export CUDA_VISIBLE_DEVICES=1

Журнал ошибок:

Не удается определить устройство cuda. Работа на устройстве 0 Ошибка сегментации (ядро сброшено)

...