在jupyter notebook中配置GPU

1. 打开Anaconda Prompt,查看虚拟环境中安装了那些kernel

jupyter kernelspec list

在这里插入图片描述
目前环境中有两个环境,一个是我们的base环境,另一个是我们新安装的tensorflowGPU版本。如果要删除新安装的GPU版本,直接按照上面的路径删除文件夹即可。接下来正式在jupyter notebook中配置GPU。

2.打开Anaconda Prompt,安装ipykernel

conda install ipykernel

在这里插入图片描述

3.接下来创建ipykernel文件

conda install -n 环境名称 ipykernel

我的GPU版本为tensorflow,所以执行以下命令即可:

conda install -n tensorflow ipykernel

在这里插入图片描述

4.进入我们的GPU环境(首先你要有这个环境)

activate 环境名称

我的GPU版本为tensorflow,所以执行以下命令即可:

conda activate tensorflow

在这里插入图片描述
出现tensoflow,则说明切换环境成功。

5.在tensorflow环境中导入kernel

python -m ipykernel install --user --name 环境名称 

我的GPU版本为tensorflow,所以执行以下命令即可:

python -m ipykernel install --user --name tensorflow 

在这里插入图片描述
此时,你可以按照上面的路径查看jupyter中是否存在tensorflow这个文件夹
在这里插入图片描述

接下来在桌面打开jupyter notebook,查看是否存在新安装的版本

在Anaconda Prompt中输入jupyter notebook
在这里插入图片描述
点击Desktop,再点击New,你就看见两个kernel
在这里插入图片描述
点击tensorflow,进入到这个GPU版本,输入以下代码检验是否可以运行

import tensorflow as tf
import os
os.environ['TF_CPP_MIN_LOG_LEVEL']='2'
a = tf.constant(1.)
b = tf.constant(2.)
print(a+b)
print('GPU:', tf.test.is_gpu_available())

在这里插入图片描述
出现红色标记说明安装成功!!!!!

  • 11
    点赞
  • 65
    收藏
    觉得还不错? 一键收藏
  • 6
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 6
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值