1.安装插件,在非虚拟环境
conda install nb_conda
conda install ipykernel
2、安装ipykernel包,在虚拟环境下安装
在Windows使用下面命令:激活环境并安装插件(这里的 Keras 是我的环境名,安装的时候换成自己的环境名即可)
activate keras
conda install ipykernel
在linux 使用下面的命令: 激活环境并安装插件
source activate keras
conda install ipykernel
安装后的效果图如下:
至此即可以愉快地使用环境中的 Keras 和 pytorch 了
Keras 及tensorflow的测试截图
pytorch测试截图:
上述的测试同时也测试了 GPU,经测试:
tensorflow 有返回 GPU 信息:
<tensorflow.python.client.session.Session at 0x2a3f4ec44a8>
pytorch GPU 测试返回 True
Reference: