如果是Anaconda环境,建议先看第9条。
- 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,成功,不再报找不到指定模块的问题。
- 但是在运行的时候提示其他错误:RuntimeError: cuDNN error: CUDNN_STATUS_EXECUTION_FAILED查找网上的解决方法,说是很可能是版本问题,我把torch卸载重装的时候,命令行给的提示是,即将下载torch……cuda90…(记不清了),我发现这里有问题,本机实际不是cuda90,而是cuda10.0。所以我回到https://pytorch.org/,并且勾选的是conda(7说明conda的安装方法可能更有效)。此外,下面的cuda选项只有9.2和10.1,我的是10.0,所以暂且勾选10.1,将提示的命令中的10.1改换成10.0。安装完成,再运行代码,ok。
- 总结:有conda的环境的话,最好是选用conda环境。在conda环境下,去pytorch官网https://pytorch.org/去找对应的命令,直接去conda prompt(管理员身份运行)输入命令安装。