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

做实验在自己电脑上没什么问题(3050ti),在借用学长电脑(3080)时却出现以上问题,主要原因可能是cuda跟显卡的兼容问题,按以下指令重新下载后问题消失。

pip install torch==1.7.1+cu110 torchvision==0.8.2+cu110 torchaudio===0.7.2 -f https://download.pytorch.org/whl/torch_stable.html
 

你可能感兴趣的:(错误类型,pytorch)