RuntimeError: nvrtc: error: invalid value for --gpu-architecture (-arch)
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.
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.
use cuda11.6 torch 1.13