报错信息:
RuntimeError: nvrtc: error: failed to open libnvrtc-builtins.so.11.1.
Make sure that libnvrtc-builtins.so.11.1 is installed correctly.
如果你是通过pip对Pytorch进行安装,会导致缺失libnvrtc-builtins.so.11.1这个文件,从而出现这个错误。
但并不需要重装torch、torchvision或者cuda,可以通过将torch的安装路径(比如我是用conda环境,路径就是anaconda3/envs/envs_name/lib/python3.7/site-packages/torch)下torch/lib中的“libnvrtc-builtins.so”文件命名为“libnvrtc-builtins.so.11.1”,注意后缀不是".so"变成了".so.11.1"。
Debug方法源自:Torchscript torch.jit.trace nvrtc: error: failed to open libnvrtc-builtins.so.11.1. · Issue #68980 · pytorch/pytorch · GitHub一次成功,分享给大家。
“All is well.”