Open dbdxnuliba opened 1 year ago
Is this resolved? I have the same issue.
I had to patch tensor2robot code to fix this
diff --git a/utils/tensorspec_utils.py b/utils/tensorspec_utils.py index 6331ec3..1f611a7 100644 --- a/utils/tensorspec_utils.py +++ b/utils/tensorspec_utils.py @@ -26,12 +26,14 @@ from six.moves import cPickle from six.moves import zip from tensor2robot.proto import t2r_pb2 import tensorflow.compat.v1 as tf +from tensorflow.python.util import nest +from tensorflow import TensorSpec as TSPEC
from google.protobuf import text_format -from tensorflow.contrib import framework as contrib_framework +#from tensorflow.contrib import framework as contrib_framework
-nest = contrib_framework.nest -TSPEC = contrib_framework.TensorSpec +#nest = contrib_framework.nest +#TSPEC = contrib_framework.TensorSpec
EXTRA_ASSETS_DIRECTORY = 'assets.extra' T2R_ASSETS_FILENAME = 't2r_assets.pbtxt'
my env :
GPU driver (google_RT1) robot@robot:/usr/local$ ls -al cuda lrwxrwxrwx 1 root root 20 Sep 20 15:10 cuda -> /usr/local/cuda-10.0
(google_RT1) robot@robot:/usr/local$ nvidia-smi Fri Dec 16 11:11:13 2022
+-----------------------------------------------------------------------------+ | NVIDIA-SMI 418.88 Driver Version: 418.88 CUDA Version: 10.1 | |-------------------------------+----------------------+----------------------+ | GPU Name Persistence-M| Bus-Id Disp.A | Volatile Uncorr. ECC | | Fan Temp Perf Pwr:Usage/Cap| Memory-Usage | GPU-Util Compute M. | |===============================+======================+======================| | 0 GeForce RTX 208... Off | 00000000:01:00.0 On | N/A | | 0% 39C P8 16W / 257W | 185MiB / 10986MiB | 5% Default | +-------------------------------+----------------------+----------------------+
+-----------------------------------------------------------------------------+ | Processes: GPU Memory | | GPU PID Type Process name Usage | |=============================================================================| | 0 1142 G /usr/lib/xorg/Xorg 183MiB | +-----------------------------------------------------------------------------+
python3:
(google_RT1) robot@robot:/usr/local$ python Python 3.8.15 (default, Nov 24 2022, 15:19:38) [GCC 11.2.0] :: Anaconda, Inc. on linux Type "help", "copyright", "credits" or "license" for more information.
Question: I has successfully configure env as your install step instruction: