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

[Muhammad Usama]

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)

I am getting this error.
I am using RTX 4090
CUDA 11.3
Nvidia driver 535 version
and pytorch is installed 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"

I have done these steps and they works fine:

python examples/1080_balls_of_solitude.py
Install the go1_gym package
In this repository, run pip install -e .

when I run python scripts/test.py, I got his error.

[Taki]

I've encountered the same issue before. It might be due to the Nvidia driver being too new. Updating to CUDA 11.6 should resolve it.

[czw1207]

use cuda11.6 torch 1.13