pytorch 之 Jupyter notebook 配置
在安装好anaconda和cuda之后,要添加虚拟环境到 Jupyter Notebook
打开anaconda,启动创建好的虚拟环境:conda activate my_pytorch (自己创建的环境名)
**需要在虚拟环境下安装 ipykernel(用于提供 Jupyter 运行所需的 IPython kernel):**
**pip install --user ipykernel**
**将我们的虚拟环境添加到 Jupyter,my_pytorch可以用自己的环境名:**
**python -m ipykernel install --user --name=my_pytorch --display-name "Python3.9 64bit ('my_pytorch': conda)"**
** 退出虚拟环境输入 jupyter kernelspec list 回车。如果没有出现列表,说明jupyter命令不能运行,添加一条环境变量:**
重新打开Anaconda Prompt,执行jupyter kernelspec list测试,如果发现列表中出现之前添加的虚拟环境,就说明成功了:
打开 jupyter notebook验证:
创建一个test.ipynb,点击kernel,点击change kernel,选择下面那个
在test.ipynb写在如下代码,并运行出现如下结果,说明可在gpu上运行