如何解决Torch 已安装,但我无法将其导入 Jupyter Notebook 的计算机视觉 Python 项目中
我正在使用检测器 2 进行计算机视觉项目。我在安装 Torch 或将其导入我的 jupyter 笔记本时遇到问题。我正在使用运行 MACOS Catalina,python3 版本为 3.8.2 的 mac,我在我的开发环境中使用 Anaconda。在屏幕截图中,它说火炬已安装,但在导入时抛出错误
解决方法
您已在 torch
/usr/local/lib/python3.9/site-packages
for python
但正如你自己所说,你是
在我的开发环境中使用 Anaconda
因此,如果您的 jupyter notebook 确实配置为使用与 anaconda 一起提供的 python 安装,那么您需要这样做
conda install pytorch torchvision torchaudio cudatoolkit=10.1 -c pytorch
注意:
-
通过运行确定您的 jupyter 实际使用的是哪个 python 解释器
import sys; print(sys.executable)
-
确保为正确的
torch
环境安装了conda
(上面的命令还会指明是哪一个) -
根据您的需要设置 cudatoolkit 版本。 official website上还有一个简单的界面,可以获取正确的命令
版权声明:本文内容由互联网用户自发贡献,该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 dio@foxmail.com 举报,一经查实,本站将立刻删除。