在Jupyter中import torch时报错的一种解决方法

在Jupyter中import torch时报错的一种解决方法

在Jupyter中import torch时报错的解决方法在使用JetPack 4.6和PyTorch 1.8的Jetson Nano开发板上，Jupyter Notebook遇到了一个特别的导入问题。
当你尝试在Jupyter Notebook中导入pytorch包时，可能会遇到不寻常的错误，但同样的代码在终端和PyCharm中却能正常运行。有人建议切换到PyCharm，但作为追求完美的人，这并不理想。
为了解决这个问题，尝试过一些解决方案，包括调整import语句的顺序，将cv2放在tensorflow和keras之前，或者在启动Jupyter Notebook前引入环境变量。然而，只有在启动前引入环境变量的方法在笔者的测试中有效，但每次都需要重复操作，略显繁琐。
经过官方信息的提示，问题的关键在于Jupyter启动了独立的Python进程，这导致引入的环境变量只对当前进程生效。为了解决这个困扰，可以在Jupyter开机启动时自动引入环境变量。
具体操作如下：检查Jupyter的开机启动配置文件（通常位于/etc/systemd/system中），找到对应的服务文件，添加以下行：
[Service]Environment="your_environment_variable=your_value"保存并重启服务，这样每次开机启动Jupyter时，它都会自动带上所需的环境变量，无需手动设置。
最后的测试结果表明，这个方法成功解决了问题，让Jupyter在import torch时不再报错。
2024-09-21