"No module" 设置 jupyter 内核后

Question

我使用 official instructions:

安装了 tensorflow

conda create -n tf-gpu tensorflow-gpu

如果我激活内核，我可以看到已安装 tensorflow。

conda activate tf-gpu
python3 -c 'import tensorflow as tf; print(tf.__version__)'

它returns以下。

2022-03-24 01:21:50.006341: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcudart.so.10.1
2.4.1

现在我想让 tf-gpu 在 jupyter notebook 中被识别为内核。

conda deactivate tf-gpu
python -m ipykernel install --user --name=tf-gpu

但是，当我启动 jupyter notebook 时

jupyter notebook

创建的 tf-gpu 笔记本没有按预期工作。

import tensorflow

Returns下面的错误

ModuleNotFoundError: No module named 'tensorflow'

为什么无法识别包裹？

Answer 1

您需要为此环境和 select jupyter notebook 中的内核创建一个新内核，如下所示：

conda activate env_name
pip install ipykernel --user
python -m ipykernel install --user --name env_name --display-name env_name

然后打开笔记本>>点击内核>>更改内核>> select内核。

如果问题仍然存在，请告诉我们。谢谢！