Isaac Gym 1.5. 4070 ti super 报错不识别 gpu arch

更换电脑后测试代码

python train.py task=Ant checkpoint=runs/Ant_26-21-14-55/nn/last_Ant_ep_500_rew__6816.49_.pth test=True num_envs=64

出现报错：

ImportError: libpython3.7m.so.1.0: cannot open shared object file: No such file or directory

参考Python joint_monkey.py ImportError: libpython3.7m.so.1.0: - Isaac Gym - NVIDIA Developer Forums

执行命令后暂时解决问题

export LD_LIBRARY_PATH=/home/blamlight/anaconda3/envs/isaacgym/lib

随后报错：

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

更新前环境使用3090，当前使用4070 ti super，已经安装了驱动，但是识别不出架构。

系统驱动：

Conda环境cuda:

/home/blamlight/anaconda3/envs/isaacgym/bin/python /home/blamlight/Documents/Github/Reinforcement-Learning/Pytorch/T_Pytorch.py 
1.8.1+cu111
True
11.1
0
1
NVIDIA GeForce RTX 4070 Ti SUPER
True
8005

Process finished with exit code 0

检查最新Log in | NVIDIA Developer后，没有出现对cuda12.1的更新。所以新建一个使用cuda12.1的conda环境。

PyTorch

name: isaacgym2
channels:
  - pytorch
  - conda-forge
  - defaults
  - nvidia
dependencies:
  - python=3.8
  - torchvision
  - pytorch-cuda=12.1
  - pyyaml>=5.3.1
  - scipy>=1.5.0
  - tensorboard>=2.2.1

随后激活环境在IsaacGym和IsaacGymEnvs中安装相关库即可。

pip install -e .

测试正常运行。