栏目分类:
子分类:
返回
终身学习网用户登录
快速导航关闭
当前搜索
当前分类
子分类
实用工具
热门搜索
终身学习网 > IT > 软件开发 > 后端开发 > Python

Pytorch学习笔记(1):import torch无法使用,就老老实实按照官方建议做就对了

Python 更新时间:发布时间: 百科书网 趣学号

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

import torch

结果提示了

Traceback (most recent call last):
  File "", line 1, in 
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中更新,欢迎访问)

转载请注明:文章转载自 www.051e.com
本文地址:http://www.051e.com/it/280360.html
我们一直用心在做
关于我们 文章归档 网站地图 联系我们

版权所有 ©2023-2025 051e.com

ICP备案号:京ICP备12030808号