引自免费微信小程序:皆我百晓生
在Jupyter Notebook或JupyterLab环境中,由于你已经创建并激活了一个名为yj_py312的conda环境,并且在这个环境中正确安装了torch库,使得torch.cuda.is_available()能够返回True。然而,当你直接在命令行中运行streamlit run web_demo_streamlit.py时,它默认使用的是系统级别的Python解释器,而非你在Jupyter中指定的conda环境。
为了解决这个问题,你需要确保streamlit run命令使用的是yj_py312环境下的Python解释器。具体操作步骤如下:
# 首先激活你的conda环境
conda activate yj_py312
# 然后使用该环境下python执行streamlit run命令
(yj_py312) which python # 检查当前环境下的Python解释器路径,确保正确
(yj_py312) python -m streamlit run /path/to/your/web_demo_streamlit.py
如果在Jupyter Notebook或JupyterLab中执行上述命令,你需要将它们写入一个shell命令单元格,并使用感叹号(!)前缀来执行shell命令:
!conda activate yj_py312
!which python
!python -m streamlit run /path/to/your/web_demo_streamlit.py
通过这种方式,streamlit run将会使用包含torch模块的conda环境来运行你的应用,从而避免ModuleNotFoundError的问题。请根据实际情况替换/path/to/your/web_demo_streamlit.py为你的.py文件的实际路径。