Skip to content

GPU Architecture Error #43

Description

@HaokunJu

Hi, great work. I try to run the code on a 5070Ti. But it issued an error related to the gpu-architecture. How can I fix it? Thanks a lot.

Traceback (most recent call last):
  File "/home/juahaokun/InterNav/LeggedLab/legged_lab/scripts/train.py", line 100, in <module>
    train()
  File "/home/juahaokun/InterNav/LeggedLab/legged_lab/scripts/train.py", line 74, in train
    env = env_class(env_cfg, args_cli.headless)
  File "/home/juahaokun/InterNav/LeggedLab/legged_lab/envs/base/base_env.py", line 78, in __init__
    self.init_buffers()
  File "/home/juahaokun/InterNav/LeggedLab/legged_lab/envs/base/base_env.py", line 127, in init_buffers
    self.init_obs_buffer()
  File "/home/juahaokun/InterNav/LeggedLab/legged_lab/envs/base/base_env.py", line 269, in init_obs_buffer
    actor_obs, _ = self.compute_current_observations()
  File "/home/juahaokun/InterNav/LeggedLab/legged_lab/envs/base/base_env.py", line 134, in compute_current_observations
    projected_gravity = robot.data.projected_gravity_b
  File "/home/juahaokun/IsaacLab/source/isaaclab/isaaclab/assets/articulation/articulation_data.py", line 776, in projected_gravity_b
    return math_utils.quat_apply_inverse(self.root_link_quat_w, self.GRAVITY_VEC_W)
RuntimeError: nvrtc: error: invalid value for --gpu-architecture (-arch)

nvrtc compilation failed: 

#define NAN __int_as_float(0x7fffffff)
#define POS_INFINITY __int_as_float(0x7f800000)
#define NEG_INFINITY __int_as_float(0xff800000)


template<typename T>
__device__ T maximum(T a, T b) {
  return isnan(a) ? a : (a > b ? a : b);
}

template<typename T>
__device__ T minimum(T a, T b) {
  return isnan(a) ? a : (a < b ? a : b);
}

extern "C" __global__
void fused_mul_sub_add(float* tv_, float* tvec_7, float* tv__, float* tt_1, float* aten_add) {
{
  float v = __ldg(tvec_7 + ((long long)(threadIdx.x) + 512ll * (long long)(blockIdx.x)) % 3ll + 2ll * ((((long long)(threadIdx.x) + 512ll * (long long)(blockIdx.x)) / 3ll) * 3ll));
  float v_1 = __ldg(tv__ + 7ll * (((long long)(threadIdx.x) + 512ll * (long long)(blockIdx.x)) / 3ll));
  float v_2 = __ldg(tt_1 + (long long)(threadIdx.x) + 512ll * (long long)(blockIdx.x));
  float v_3 = __ldg(tv_ + (long long)(threadIdx.x) + 512ll * (long long)(blockIdx.x));
  aten_add[(long long)(threadIdx.x) + 512ll * (long long)(blockIdx.x)] = (v - v_1 * v_2) + v_3;
}
}

Activity

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Projects

    No projects

      Milestone

      No milestone

      Relationships

      None yet

      Development

      No branches or pull requests

      Issue actions