安装Pytorch No module named ‘tools.nnwrap解决办法

问题描述

使用pycharm进行安装或使用pip进行安装而没有指定版本和CUDA版本，如执行pip install torch的时候，会出现这个错误。原因是python并不知道你使用的是什么版本的CUDA，或者你根本没有安装CUDA。

注意：安装pytorch之前需要先安装CUDA，可以参考这个链接安装CUDA。

解决办法

1、 首先安装CUDA（安装过的请跳过）

安装过程不在这里赘述，详情可以参考上面的链接。

2、开始安装Pytorch

点击这里进入Pytorch官网，首页下拉可以看到选择Pytorch的界面，如下图：

3、复制下面的pip语句

4、安装pytorch

打开pycahrm，或者打开CMD都可以，前提是你的默认解释器和当前使用的解释器是同一个，个人推荐使用pycahrm的，可以保证你使用的是当前的解释器。

1. 打开终端：

2. 粘贴我们刚才复制的语句

3. 等待安装成功

由于我已经安装完成，所以看不到安装过程，但是是可以安装成功的。

至此，问题解决！！！