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上运行

posted @ 2022-04-15 20:18  一只皮π虾  阅读(613)  评论(0编辑  收藏  举报