如果是Anaconda环境,建议先看第7条。
- pip直接安装报错
尝试去官网查看https://pytorch.org/,点击get start,根据自己的实际情况勾选信息,得到提示的代码,复制到命令行运行即可 - 找不到指定的模块的问题
可能是环境变量的问题,按照说明添加了,还是出错,所以应该不是环境变量的问题。
也有可能是numpy版本的问题,尝试着更新numpy,第一次使用的是whl安装方式。 - whl版本问题
从https://www.lfd.uci.edu/~gohlke/pythonlibs/#numpy中下载whl文件,安装报错:
提示这个版本在本平台不支持,所以应该先查一下可以支持的版本,应该进入python,import pip,print(pip.pep425tags.get_supported()),查看结果,据此下载对应版本。 - –user的权限问题
可能是anaconda的保护性,更新numpy,先是去删除旧版numpy,whl方式下,提示某个文件夹访问权限。因此,直接换用upgrade的方式,依旧受阻。需要在代码里添加–user
成功安装。 - conda与pip的版本共存问题
不过,这个pip的安装目录在另外一个文件夹。而conda本身环境中就有一个同名的,而且版本不同的以来库。因此,使用conda的库的时候,就会优先使用conda中的依赖,而不是这次的pip安装的依赖。所以我们需要换conda安装,将需要依赖的库直接放入conda环境中。 - conda更新numpy成功之后,依旧提示原来的错误,不知如何解决。
- conda prompt右键,管理员身份打开(不然会提示写入权限问题),输入conda install pytorch torchvision cudatoolkit=9.0 -c pytorch,成功,不再报找不到指定模块的问题。