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

问题描述

在使用pytorch3d的时候，可以正常的import，但是在执行错误的使用就会报，nvrtc: error: invalid value for --gpu-architecture (-arch)，的错误，图片如下：

我的环境是：

显卡： 4090

torch: 1.12.1

python: 3.10

解决办法：

我是参考了 https://github.com/THUDM/ChatGLM-6B/issues/593 这个链接，就是将torch升级到13.0或者以上的版本，我升级到了13.1

pip install torch==1.13.1+cu117 torchvision==0.14.1+cu117 torchaudio==0.13.1 --extra-index-url https://download.pytorch.org/whl/cu117

##

祝你好运

# 有问题可以进群聊聊

614809646  qq群->数字人和tts,运维 等等