I am currently using the following settings:
Driver Version : 525.116.04
CUDA: 11.8
torch: 1.8.1
when I try to run
python train.py task=Ant
It returns the following error log:
Traceback (most recent call last):
File "train.py", line 194, in launch_rlg_hydra
'sigma': cfg.sigma if cfg.sigma != '' else None
File "/home/tc/anaconda3/envs/rlgpu/lib/python3.7/site-packages/rl_games/torch_runner.py", line 121, in run
self.run_train(args)
File "/home/tc/anaconda3/envs/rlgpu/lib/python3.7/site-packages/rl_games/torch_runner.py", line 102, in run_train
agent.train()
File "/home/tc/anaconda3/envs/rlgpu/lib/python3.7/site-packages/rl_games/common/a2c_common.py", line 1226, in train
step_time, play_time, update_time, sum_time, a_losses, c_losses, b_losses, entropies, kls, last_lr, lr_mul = self.train_epoch()
File "/home/tc/anaconda3/envs/rlgpu/lib/python3.7/site-packages/rl_games/common/a2c_common.py", line 1090, in train_epoch
batch_dict = self.play_steps()
File "/home/tc/anaconda3/envs/rlgpu/lib/python3.7/site-packages/rl_games/common/a2c_common.py", line 674, in play_steps
self.obs, rewards, self.dones, infos = self.env_step(res_dict['actions'])
File "/home/tc/anaconda3/envs/rlgpu/lib/python3.7/site-packages/rl_games/common/a2c_common.py", line 495, in env_step
obs, rewards, dones, infos = self.vec_env.step(actions)
File "/home/tc/RL_Project/IsaacGymEnvs/isaacgymenvs/utils/rlgames_utils.py", line 246, in step
return self.env.step(actions)
File "/home/tc/RL_Project/IsaacGymEnvs/isaacgymenvs/tasks/base/vec_task.py", line 355, in step
self.post_physics_step()
File "/home/tc/RL_Project/IsaacGymEnvs/isaacgymenvs/tasks/ant.py", line 288, in post_physics_step
self.reset_idx(env_ids)
File "/home/tc/RL_Project/IsaacGymEnvs/isaacgymenvs/tasks/ant.py", line 252, in reset_idx
positions = torch_rand_float(-0.2, 0.2, (len(env_ids), self.num_dof), device=self.device)
RuntimeError: The following operation failed in the TorchScript interpreter.
Traceback of TorchScript (most recent call last):
RuntimeError: nvrtc: error: invalid value for --gpu-architecture (-arch)
Can someone help me with this please?