Skip to content

GPU Architecture Error #43

@HaokunJu

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;
}
}

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