RuntimeError: nvrtc: error: invalid value for --gpu-architecture (-arch)

[tula-liuxin]

legged_gym) leo@leo-pc:~/legged_gym$ python legged_gym/scripts/train.py --task=anymal_c_flat
Importing module 'gym_38' (/home/leo/isaacgym/python/isaacgym/_bindings/linux-x86_64/gym_38.so)
Setting GYM_USD_PLUG_INFO_PATH to /home/leo/isaacgym/python/isaacgym/_bindings/linux-x86_64/usd/plugInfo.json
PyTorch version 1.10.0+cu113
Device count 1
/home/leo/isaacgym/python/isaacgym/_bindings/src/gymtorch
Using /home/leo/.cache/torch_extensions/py38_cu113 as PyTorch extensions root...
Emitting ninja build file /home/leo/.cache/torch_extensions/py38_cu113/gymtorch/build.ninja...
Building extension module gymtorch...
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 gymtorch...
Setting seed: 1
Not connected to PVD
+++ Using GPU PhysX
Physics Engine: PhysX
Physics Device: cuda:0
GPU Pipeline: enabled
/home/leo/miniforge3/envs/legged_gym/lib/python3.8/site-packages/torch/functional.py:445: UserWarning: torch.meshgrid: in an upcoming release, it will be required to pass the indexing argument. (Triggered internally at ../aten/src/ATen/native/TensorShape.cpp:2157.)
return _VF.meshgrid(tensors, **kwargs) # type: ignore[attr-defined]
Traceback (most recent call last):
File "legged_gym/scripts/train.py", line 47, in
train(args)
File "legged_gym/scripts/train.py", line 41, in train
env, env_cfg = task_registry.make_env(name=args.task, args=args)
File "/home/leo/legged_gym/legged_gym/utils/task_registry.py", line 97, in make_env
env = task_class( cfg=env_cfg,
File "/home/leo/legged_gym/legged_gym/envs/anymal_c/anymal.py", line 49, in init
super().init(cfg, sim_params, physics_engine, sim_device, headless)
File "/home/leo/legged_gym/legged_gym/envs/base/legged_robot.py", line 71, in init
super().init(self.cfg, sim_params, physics_engine, sim_device, headless)
File "/home/leo/legged_gym/legged_gym/envs/base/base_task.py", line 84, in init
self.create_sim()
File "/home/leo/legged_gym/legged_gym/envs/base/legged_robot.py", line 244, in create_sim
self._create_envs()
File "/home/leo/legged_gym/legged_gym/envs/base/legged_robot.py", line 675, in _create_envs
pos[:2] += torch_rand_float(-1., 1., (2,1), device=self.device).squeeze(1)
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)

[Cdfghglz]

Your gpu may be too new, not supporting cuda 11.3 - this was my issue with a 4060 recently.

[Ke-Wang1017]

Actually newer version of Pytorch and torchvision is working fine for me. I suggest you check your Cuda version with nvidia-smi and install the corresponding version. For me the cuda is 12.1 and I installed the newest pytorch1

[tula-liuxin]

python legged_gym/scripts/train.py --task=anymal_c_flat

Thank you, it helps! I didn't know the change the pytorch version given my cuda version, and it works well now.

[tula-liuxin]

Your gpu may be too new, not supporting cuda 11.3 - this was my issue with a 4060 recently.

Thank your comments! I found it's about the imcompatibility between the pytorch and cuda.

[tula-liuxin]

Solved by:
pip3 uninstall torch torchvision torchaudio
pip3 install torch torchvision torchaudio -f https://download.pytorch.org/whl/cu123/torch_stable.html #My cuda version is 12.3

[KevinGrey]

cuda 12.0
RTX4090
ubuntu22.04
python3.7.12
torch 1.8
then
I uninstalled torch
pip uninstall torch torchvision torchaudio
and
installed torch 1.13
fixed

[Andy-xiong6]

cuda 12.2
RTX4060 TI
ubuntu20.04
python3.8
torch 1.10
then
I uninstalled torch
pip uninstall torch torchvision torchaudio
and
installed torch 2.3.0+cu121
fixed

[asibatian]

RTX4060 Ti
Ubuntu20.04
Python3.8
At first, I installed cuda11.3 and install torch using pip3 install torch==1.10.0+cu113 torchvision==0.11.1+cu113 torchaudio==0.10.0+cu113 -f https://download.pytorch.org/whl/cu113/torch_stable.html (This version I works well on RTX3090)
It failed.
Then, I uninstall cuda 11.3 and install cuda12.1
uninstall torch using pip uninstall torch torchvision torchaudio
install torch using pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu121
torch.__version__: 2.3.1+cu121
It works