How to solve the following error?

[Davidzhu5]

D:\ai\DecisionRWKV-main\venv\Scripts\python.exe "C:/Program Files/JetBrains/PyCharm 2023.1.3/plugins/python/helpers/pydev/pydevconsole.py" --mode=client --host=127.0.0.1 --port=5741
import sys; print('Python %s on %s' % (sys.version, sys.platform))
sys.path.extend(['D:\ai\DecisionRWKV-main'])
PyDev console: starting.
Python 3.10.6 (tags/v3.10.6:9c7b4bd, Aug 1 2022, 21:53:49) [MSC v.1932 64 bit (AMD64)] on win32
runfile('D:\ai\DecisionRWKV-main\D4RL\dRWKV6.py', wdir='D:\ai\DecisionRWKV-main\D4RL')
Warning: Flow failed to import. Set the environment variable D4RL_SUPPRESS_IMPORT_ERROR=1 to suppress this message.
No module named 'flow'
Warning: CARLA failed to import. Set the environment variable D4RL_SUPPRESS_IMPORT_ERROR=1 to suppress this message.
No module named 'agents'
pybullet build time: Sep 8 2024 13:19:08
Using C:\Users\zz\AppData\Local\torch_extensions\torch_extensions\Cache\py310_cu121 as PyTorch extensions root...
Detected CUDA files, patching ldflags
Emitting ninja build file C:\Users\zz\AppData\Local\torch_extensions\torch_extensions\Cache\py310_cu121\wkv6\build.ninja...
D:\ai\DecisionRWKV-main\venv\lib\site-packages\torch\utils\cpp_extension.py:1965: UserWarning: TORCH_CUDA_ARCH_LIST is not set, all archs for visible cards are included for compilation.
If this is not desired, please set os.environ['TORCH_CUDA_ARCH_LIST'].
warnings.warn(
Building extension module wkv6...
Allowing ninja to set a default number of workers... (overridable by setting the environment variable MAX_JOBS=N)
[1/2] C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v12.1\bin\nvcc --generate-dependencies-with-compile --dependency-output wkv6_cuda.cuda.o.d -Xcudafe --diag_suppress=dll_interface_conflict_dllexport_assumed -Xcudafe --diag_suppress=dll_interface_conflict_none_assumed -Xcudafe --diag_suppress=field_without_dll_interface -Xcudafe --diag_suppress=base_class_has_different_dll_interface -Xcompiler /EHsc -Xcompiler /wd4068 -Xcompiler /wd4067 -Xcompiler /wd4624 -Xcompiler /wd4190 -Xcompiler /wd4018 -Xcompiler /wd4275 -Xcompiler /wd4267 -Xcompiler /wd4244 -Xcompiler /wd4251 -Xcompiler /wd4819 -Xcompiler /MD -DTORCH_EXTENSION_NAME=wkv6 -DTORCH_API_INCLUDE_EXTENSION_H -ID:\ai\DecisionRWKV-main\venv\lib\site-packages\torch\include -ID:\ai\DecisionRWKV-main\venv\lib\site-packages\torch\include\torch\csrc\api\include -ID:\ai\DecisionRWKV-main\venv\lib\site-packages\torch\include\TH -ID:\ai\DecisionRWKV-main\venv\lib\site-packages\torch\include\THC "-IC:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v12.1\include" -ID:\Python310\Include -D_GLIBCXX_USE_CXX11_ABI=0 -D__CUDA_NO_HALF_OPERATORS__ -D__CUDA_NO_HALF_CONVERSIONS__ -D__CUDA_NO_BFLOAT16_CONVERSIONS__ -D__CUDA_NO_HALF2_OPERATORS__ --expt-relaxed-constexpr -gencode=arch=compute_89,code=compute_89 -gencode=arch=compute_89,code=sm_89 -std=c++17 -res-usage --use_fast_math -O3 -Xptxas=-O3 --extra-device-vectorization -D_N_=64 -D_T_=256 -c D:\ai\DecisionRWKV-main\D4RL\cuda\wkv6_cuda.cu -o wkv6_cuda.cuda.o
ptxas info : 1 bytes gmem
ptxas info : Compiling entry function 'Z19kernel_backward_222IfEviiiiPKT_S2_S2_PKfS2_S2_PS0' for 'sm_89'
ptxas info : Function properties for Z19kernel_backward_222IfEviiiiPKT_S2_S2_PKfS2_S2_PS0
1016 bytes stack frame, 0 bytes spill stores, 0 bytes spill loads
ptxas info : Used 96 registers, 512 bytes smem, 424 bytes cmem[0]
ptxas info : Compiling entry function 'Z19kernel_backward_111IfEviiiiPKT_S2_S2_PKfS2_S2_PS0_S5_S5_S5' for 'sm_89'
ptxas info : Function properties for Z19kernel_backward_111IfEviiiiPKT_S2_S2_PKfS2_S2_PS0_S5_S5_S5
0 bytes stack frame, 0 bytes spill stores, 0 bytes spill loads
ptxas info : Used 128 registers, 1536 bytes smem, 448 bytes cmem[0]
ptxas info : Compiling entry function 'Z14kernel_forwardIfEviiiiPKT_S2_S2_PKfS2_PS0' for 'sm_89'
ptxas info : Function properties for Z14kernel_forwardIfEviiiiPKT_S2_S2_PKfS2_PS0
0 bytes stack frame, 0 bytes spill stores, 0 bytes spill loads
ptxas info : Used 100 registers, 1024 bytes smem, 416 bytes cmem[0]
wkv6_cuda.cu
tmpxft_00005678_00000000-10_wkv6_cuda.cudafe1.cpp
[2/2] "C:\Program Files (x86)\Microsoft Visual Studio\2019\BuildTools\VC\Tools\MSVC\14.29.30133\bin\Hostx64\x64/link.exe" wkv6_op.o wkv6_cuda.cuda.o /nologo /DLL c10.lib c10_cuda.lib torch_cpu.lib torch_cuda.lib -INCLUDE:?warp_size@cuda@at@@yahxz torch.lib /LIBPATH:D:\ai\DecisionRWKV-main\venv\lib\site-packages\torch\lib torch_python.lib /LIBPATH:D:\Python310\libs "/LIBPATH:C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v12.1\lib\x64" cudart.lib /out:wkv6.pyd
���ڴ����� wkv6.lib �Ͷ��� wkv6.exp
Loading extension module wkv6...
wandb: Currently logged in as: 431113249 (431113249-piabia). Use wandb login --relogin to force relogin
wandb: Tracking run with wandb version 0.17.9
wandb: Run data is saved locally in D:\ai\DecisionRWKV-main\D4RL\wandb\run-20240914_105722-a695536d-60f4-44ca-8536-42da60f8144c
wandb: Run wandb offline to turn off syncing.
wandb: Syncing run DRWKV6-halfcheetah-medium-v2-e495441f
wandb: View project at https://wandb.ai/431113249-piabia/D-RWKV
wandb: View run at https://wandb.ai/431113249-piabia/D-RWKV/runs/a695536d-60f4-44ca-8536-42da60f8144c
wandb: WARNING Calling wandb.run.save without any arguments is deprecated.Changes to attributes are automatically persisted.
D:\ai\DecisionRWKV-main\venv\lib\site-packages\gym\spaces\box.py:84: UserWarning: WARN: Box bound precision lowered by casting to float32
logger.warn(f"Box bound precision lowered by casting to {self.dtype}")
load datafile: 100%|██████████| 21/21 [00:01<00:00, 17.44it/s]
Processing trajectories: 100%|██████████| 1000000/1000000 [00:01<00:00, 591168.19it/s]
Total parameters: 3600518
Traceback (most recent call last):
File "C:\Program Files\JetBrains\PyCharm 2023.1.3\plugins\python\helpers\pydev\pydevconsole.py", line 364, in runcode
coro = func()
File "", line 1, in
File "C:\Program Files\JetBrains\PyCharm 2023.1.3\plugins\python\helpers\pydev_pydev_bundle\pydev_umd.py", line 198, in runfile
pydev_imports.execfile(filename, global_vars, local_vars) # execute the script
File "C:\Program Files\JetBrains\PyCharm 2023.1.3\plugins\python\helpers\pydev_pydev_imps_pydev_execfile.py", line 18, in execfile
exec(compile(contents+"\n", file, 'exec'), glob, loc)
File "D:\ai\DecisionRWKV-main\D4RL\dRWKV6.py", line 691, in
trainConfig(env,seed)
File "D:\ai\DecisionRWKV-main\D4RL\dRWKV6.py", line 682, in trainConfig
train()
File "D:\ai\DecisionRWKV-main\venv\lib\site-packages\pyrallis\argparsing.py", line 158, in wrapper_inner
response = fn(cfg, *args, **kwargs)
File "D:\ai\DecisionRWKV-main\D4RL\dRWKV6.py", line 607, in train
trainloader_iter = iter(trainloader)
File "D:\ai\DecisionRWKV-main\venv\lib\site-packages\torch\utils\data\dataloader.py", line 440, in iter
return self._get_iterator()
File "D:\ai\DecisionRWKV-main\venv\lib\site-packages\torch\utils\data\dataloader.py", line 388, in _get_iterator
return _MultiProcessingDataLoaderIter(self)
File "D:\ai\DecisionRWKV-main\venv\lib\site-packages\torch\utils\data\dataloader.py", line 1038, in init
w.start()
File "D:\Python310\lib\multiprocessing\process.py", line 121, in start
self._popen = self._Popen(self)
File "D:\Python310\lib\multiprocessing\context.py", line 224, in _Popen
return _default_context.get_context().Process._Popen(process_obj)
File "D:\Python310\lib\multiprocessing\context.py", line 336, in _Popen
return Popen(process_obj)
File "D:\Python310\lib\multiprocessing\popen_spawn_win32.py", line 93, in init
reduction.dump(process_obj, to_child)
File "D:\Python310\lib\multiprocessing\reduction.py", line 60, in dump
ForkingPickler(file, protocol).dump(obj)
_pickle.PicklingError: Can't pickle <class 'main.SequenceDataset'>: attribute lookup SequenceDataset on main failed
Traceback (most recent call last):
File "C:\Program Files\JetBrains\PyCharm 2023.1.3\plugins\python\helpers\pydev\pydevconsole.py", line 364, in runcode
coro = func()
File "", line 1, in
File "C:\Program Files\JetBrains\PyCharm 2023.1.3\plugins\python\helpers\pydev_pydev_bundle\pydev_umd.py", line 198, in runfile
pydev_imports.execfile(filename, global_vars, local_vars) # execute the script
File "C:\Program Files\JetBrains\PyCharm 2023.1.3\plugins\python\helpers\pydev_pydev_imps_pydev_execfile.py", line 18, in execfile
exec(compile(contents+"\n", file, 'exec'), glob, loc)
File "D:\ai\DecisionRWKV-main\D4RL\dRWKV6.py", line 691, in
trainConfig(env,seed)
File "D:\ai\DecisionRWKV-main\D4RL\dRWKV6.py", line 682, in trainConfig
train()
File "D:\ai\DecisionRWKV-main\venv\lib\site-packages\pyrallis\argparsing.py", line 158, in wrapper_inner
response = fn(cfg, *args, **kwargs)
File "D:\ai\DecisionRWKV-main\D4RL\dRWKV6.py", line 607, in train
trainloader_iter = iter(trainloader)
File "D:\ai\DecisionRWKV-main\venv\lib\site-packages\torch\utils\data\dataloader.py", line 440, in iter
return self._get_iterator()
File "D:\ai\DecisionRWKV-main\venv\lib\site-packages\torch\utils\data\dataloader.py", line 388, in _get_iterator
return _MultiProcessingDataLoaderIter(self)
File "D:\ai\DecisionRWKV-main\venv\lib\site-packages\torch\utils\data\dataloader.py", line 1038, in init
w.start()
File "D:\Python310\lib\multiprocessing\process.py", line 121, in start
self._popen = self._Popen(self)
File "D:\Python310\lib\multiprocessing\context.py", line 224, in _Popen
return _default_context.get_context().Process._Popen(process_obj)
File "D:\Python310\lib\multiprocessing\context.py", line 336, in _Popen
return Popen(process_obj)
File "D:\Python310\lib\multiprocessing\popen_spawn_win32.py", line 93, in init
reduction.dump(process_obj, to_child)
File "D:\Python310\lib\multiprocessing\reduction.py", line 60, in dump
ForkingPickler(file, protocol).dump(obj)
_pickle.PicklingError: Can't pickle <class 'main.SequenceDataset'>: attribute lookup SequenceDataset on main failed
Warning: Flow failed to import. Set the environment variable D4RL_SUPPRESS_IMPORT_ERROR=1 to suppress this message.
No module named 'flow'
2024-09-14 10:57:32.379071: I tensorflow/core/util/port.cc:153] oneDNN custom operations are on. You may see slightly different numerical results due to floating-point round-off errors from different computation orders. To turn them off, set the environment variable TF_ENABLE_ONEDNN_OPTS=0.
2024-09-14 10:57:33.068135: I tensorflow/core/util/port.cc:153] oneDNN custom operations are on. You may see slightly different numerical results due to floating-point round-off errors from different computation orders. To turn them off, set the environment variable TF_ENABLE_ONEDNN_OPTS=0.
pybullet build time: Sep 8 2024 13:19:08
current_dir=D:\ai\DecisionRWKV-main\venv\lib\site-packages\pybullet_envs\bullet
Warning: CARLA failed to import. Set the environment variable D4RL_SUPPRESS_IMPORT_ERROR=1 to suppress this message.
No module named 'agents.navigation'
Using C:\Users\zz\AppData\Local\torch_extensions\torch_extensions\Cache\py310_cu121 as PyTorch extensions root...
Detected CUDA files, patching ldflags
Emitting ninja build file C:\Users\zz\AppData\Local\torch_extensions\torch_extensions\Cache\py310_cu121\wkv6\build.ninja...
D:\ai\DecisionRWKV-main\venv\lib\site-packages\torch\utils\cpp_extension.py:1965: UserWarning: TORCH_CUDA_ARCH_LIST is not set, all archs for visible cards are included for compilation.
If this is not desired, please set os.environ['TORCH_CUDA_ARCH_LIST'].
warnings.warn(
Building extension module wkv6...
Allowing ninja to set a default number of workers... (overridable by setting the environment variable MAX_JOBS=N)
ninja: no work to do.
Loading extension module wkv6...
Traceback (most recent call last):
File "", line 1, in
File "D:\Python310\lib\multiprocessing\spawn.py", line 116, in spawn_main
exitcode = _main(fd, parent_sentinel)
File "D:\Python310\lib\multiprocessing\spawn.py", line 126, in _main
self = reduction.pickle.load(from_parent)
EOFError: Ran out of input