admin管理员组

文章数量:1663034

问题描述

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

我的环境是:

显卡: 4090

torch: 1.12.1

python: 3.10

解决办法:

我是参考了 https://github/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/whl/cu117

##

祝你好运

有问题可以交流呀

concat("求求QUN", "61480", "9646")

本文标签: InvalidErrornvrtcArchArchitecture