安装了Pytorch,但是当执行Python的时候,输入

import torch

结果提示了

Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
ModuleNotFoundError: No module named 'pytorch'

看到了这个问题,第一件事可能是因为python的版本可能不适配。在这种情况下,首先输入

exit()

退出不执行代码的状态,随后,在此情况下,检查一下Python版本

python -v

如果能够查到Python的版本,再配置环境。比如我的Python版本是3.9,那么后面就要把版本号换成3.9。

conda create -n env_pytorch python=3.9

在执行这条规定,重新输入import torch,仍然没效果。

之前有人说,如果把Python的版本号退回到3.7及以前,就能够有效了。但是我试完,仍然是没有效果的。

突然想起来,因为我是AMD的架构,在操作系统不是Linux的情况下,只能选择CPU版本。走投无路的我终于认了命,按照单纯有CPU的情况执行命令吧。

conda install pytorch torchvision torchaudio cpuonly -c pytorch

果不其然,在执行这一指令后,再输入Import torch,结果是这样的。

虽然启动速度有些慢,但是还是正常启动了。

所以,老老实实地听话最稳妥。如果你的架构是Intel+英伟达,且有独立显卡。那就可以按照官方最佳的推荐安装方式了。否则,就用CPU版本的,也不错。

(本系列文章同步在本人博客www.im-zu.cn中更新,欢迎访问)

Logo

华为开发者空间,是为全球开发者打造的专属开发空间,汇聚了华为优质开发资源及工具,致力于让每一位开发者拥有一台云主机,基于华为根生态开发、创新。

更多推荐