大概率是环境出了问题。首先进入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
大概率是环境出了问题。首先进入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