我使用official instructions安装了张流:
conda create -n tf-gpu tensorflow-gpu
如果我激活内核,我可以看到tensorflow已经安装。
conda activate tf-gpu
python3 -c 'import tensorflow as tf; print(tf.__version__)'
它返回以下内容。
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
创建的tf-gpu
笔记本无法按预期工作。
import tensorflow
返回以下错误
ModuleNotFoundError: No module named 'tensorflow'
为什么包未被识别?
2条答案
按热度按时间nlejzf6q1#
您需要为这个env创建一个新的内核,并从jupyter notebook中选择内核,如下所示:
然后打开笔记本〉〉点击内核〉〉更改内核〉〉选择内核。
如果问题仍然存在,请告诉我们。谢谢!
6fe3ivhb2#
或者,您也可以在安装后使用
ipython kernel
代替ipykernel
,如下所示:ipython kernel install --user --name env_name --display-name env_name