解决nvrtc: error: invalid value for --gpu-architecture (-arch)
这个链接,就是将torch升级到13.0或者以上的版本,我升级到了13.1。
·
问题描述
在使用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
##
祝你好运
有问题可以交流呀
concat("求求QUN", "61480", "9646")



更多推荐
所有评论(0)