search

Improbable-AI / walk-these-ways

Sim-to-real RL training and deployment tools for the Unitree Go1 robot.
https://gmargo11.github.io/walk-these-ways/
Other
488 stars 129 forks source link

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

Open usama-x930 opened 4 months ago

usama-x930 commented 4 months ago

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.

147mmd commented 3 months ago

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.