ImportError: libc10_cuda.so: cannot open shared object file: No such file or directory

大概率是环境出了问题。首先进入python环境,检查torch好不好使:

import torch
torch.cuda.is_available()

如果为False的话,就按工程要求重装一下(一般install.md会给)。比如:

pip install torch==1.6.0+cu101 torchvision==0.7.0+cu101 -f https://download.pytorch.org/whl/torch_stable.html

你可能感兴趣的:(G2S工程记录,python,深度学习,pytorch)