hcnoh / gail-pytorch Goto Github PK

ml/lib/python3.9/site-packages/gym/envs/registration.py", line 251, in _check_version_exists
    raise error.DeprecatedEnv(
gym.error.DeprecatedEnv: Environment version v0 for `Pendulum` is deprecated. Please use `Pendulum-v1` instead.

python train.py --env_name=CartPole-v1

  File "gail-pytorch/models/nets.py", line 128, in act
    state = FloatTensor(state)
ValueError: expected sequence of length 4 at dim 1 (got 0)

python3 train.py --env_name=BipedalWalker-v3
/home/dl/GAIL/GAIL/lib/python3.8/site-packages/torch/cuda/__init__.py:146: UserWarning: 
NVIDIA GeForce RTX 3090 with CUDA capability sm_86 is not compatible with the current PyTorch installation.
The current PyTorch install supports CUDA capabilities sm_37 sm_50 sm_60 sm_70.
If you want to use the NVIDIA GeForce RTX 3090 GPU with PyTorch, please check the instructions at https://pytorch.org/get-started/locally/

  warnings.warn(incompatible_device_warn.format(device_name, capability, " ".join(arch_list), device_name))
Traceback (most recent call last):
  File "train.py", line 98, in <module>
    main(**vars(args))
  File "train.py", line 54, in main
    expert = Expert(
  File "/home/dl/GAIL/gail-pytorch/models/nets.py", line 120, in __init__
    self.pi = PolicyNetwork(self.state_dim, self.action_dim, self.discrete)
  File "/home/dl/GAIL/gail-pytorch/models/nets.py", line 18, in __init__
    Linear(state_dim, 50),
  File "/home/dl/GAIL/GAIL/lib/python3.8/site-packages/torch/nn/modules/linear.py", line 101, in __init__
    self.reset_parameters()
  File "/home/dl/GAIL/GAIL/lib/python3.8/site-packages/torch/nn/modules/linear.py", line 107, in reset_parameters
    init.kaiming_uniform_(self.weight, a=math.sqrt(5))
  File "/home/dl/GAIL/GAIL/lib/python3.8/site-packages/torch/nn/init.py", line 412, in kaiming_uniform_
    return tensor.uniform_(-bound, bound)
RuntimeError: CUDA error: no kernel image is available for execution on the device
CUDA kernel errors might be asynchronously reported at some other API call,so the stacktrace below might be incorrect.
For debugging consider passing CUDA_LAUNCH_BLOCKING=1.

$ python3 train.py --env_name=BipedalWalker-v3
Traceback (most recent call last):
  File "/home/dl/GAIL/GAIL/lib/python3.8/site-packages/torch/__init__.py", line 172, in _load_global_deps
    ctypes.CDLL(lib_path, mode=ctypes.RTLD_GLOBAL)
  File "/usr/lib/python3.8/ctypes/__init__.py", line 373, in __init__
    self._handle = _dlopen(self._name, mode)
OSError: /home/dl/GAIL/GAIL/lib/python3.8/site-packages/torch/lib/../../nvidia/cublas/lib/libcublas.so.11: undefined symbol: cublasLtGetStatusString, version libcublasLt.so.11

During handling of the above exception, another exception occurred:

Traceback (most recent call last):
  File "train.py", line 6, in <module>
    import torch
  File "/home/dl/GAIL/GAIL/lib/python3.8/site-packages/torch/__init__.py", line 217, in <module>
    _load_global_deps()
  File "/home/dl/GAIL/GAIL/lib/python3.8/site-packages/torch/__init__.py", line 178, in _load_global_deps
    _preload_cuda_deps()
  File "/home/dl/GAIL/GAIL/lib/python3.8/site-packages/torch/__init__.py", line 158, in _preload_cuda_deps
    ctypes.CDLL(cublas_path)
  File "/usr/lib/python3.8/ctypes/__init__.py", line 373, in __init__
    self._handle = _dlopen(self._name, mode)
OSError: /home/dl/GAIL/GAIL/lib/python3.8/site-packages/nvidia/cublas/lib/libcublas.so.11: undefined symbol: cublasLtGetStatusString, version libcublasLt.so.11