栏目分类:
子分类:
返回
名师互学网用户登录
快速导航关闭
当前搜索
当前分类
子分类
实用工具
热门搜索
名师互学网 > IT > 软件开发 > 后端开发 > Python

在jupyter notebook中配置GPU

Python 更新时间: 发布时间: IT归档 最新发布 模块sitemap 名妆网 法律咨询 聚返吧 英语巴士网 伯小乐 网商动力

在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())


出现红色标记说明安装成功!!!!!

转载请注明:文章转载自 www.mshxw.com
本文地址:https://www.mshxw.com/it/840181.html
我们一直用心在做
关于我们 文章归档 网站地图 联系我们

版权所有 (c)2021-2022 MSHXW.COM

ICP备案号:晋ICP备2021003244-6号