seglink
seglink copied to clipboard
import error
Dear Deng: I'm sorry I am a newer for TensorFlow framework and get an error of following:
++ set -e
++ export CUDA_VISIBLE_DEVICES=0
++ CUDA_VISIBLE_DEVICES=0
++ CHECKPOINT_PATH=/home/jin/zs/seglink/seglink-512/
++ DATASET_DIR=/home/jin/data/ch4/ch4_test_imgs
++ python test_seglink.py --checkpoint_path=/home/jin/zs/seglink/seglink-512/ --gpu_memory_fraction=-1 --seg_conf_threshold=0.8 --link_conf_threshold=0.5 --dataset_dir=/home/jin/data/ch4/ch4_test_imgs
I tensorflow/stream_executor/dso_loader.cc:111] successfully opened CUDA library libcublas.so locally
I tensorflow/stream_executor/dso_loader.cc:111] successfully opened CUDA library libcudnn.so locally
I tensorflow/stream_executor/dso_loader.cc:111] successfully opened CUDA library libcufft.so locally
I tensorflow/stream_executor/dso_loader.cc:111] successfully opened CUDA library libcuda.so.1 locally
I tensorflow/stream_executor/dso_loader.cc:111] successfully opened CUDA library libcurand.so locally
Traceback (most recent call last):
File "test_seglink.py", line 9, in
I will appreciate it very much! Thanks!
What your tensorflow version? 1.1.0 should be used, higher versions should be OK too, but I am not sure.