No module named ‘torch’
No module named ‘tools.nnwrap’
这两个问题一般同时出现,原因都是安装torch是出现问题。
重新安装torch及其相关第三方库即可解决(官网获取命令参数)
PyTorch官网:https://pytorch.org/
1、pip 安装
pip install torch===1.7.0 torchvision===0.8.1 torchaudio===0.7.0 -f https://download.pytorch.org/whl/torch_stable.html```
2、conda(10.2)安装
conda install pytorch torchvision torchaudio cudatoolkit=10.2 -c pytorch