深度学习入门之Tensorflow安装、keras踩坑总结(二)——tensorflow指定GPU、Jupyter notebook切换内核、显存释放等

本文介绍了Tensorflow的安装方法,特别是通过清华大学开源软件镜像站加速下载。在Jupyter Notebook中使用Keras Tensorflow时,讨论了如何解决显存不释放和指定GPU使用的问题,包括设置`CUDA_VISIBLE_DEVICES`和`gpu_options`。同时,文章还讲述了如何在Jupyter Notebook中切换anaconda虚拟环境的内核,以便在不同环境中运行程序。

在上篇文章中,我们总结了一些在Theano安装使用过程中遇到的问题以及解决办法,接下来我们主要说一下Tensorflow。

1. Tensorflow安装

Tensorflow如果直接使用pip命令的话,可能十分耗时,下载速度非常慢,还很容易中断,所以大家可以使用清华大学的开源软件镜像站进行下载https://mirrors.tuna.tsinghua.edu.cn/help/tensorflow/。python3.5版本的话,可以选择Tensorflow1.4.0版本啦,其他版本应该也有可以的,这个版本是一个师弟推荐我使用的,所以我就直接使用了这个啦~  

我们可以根据自己用CPU还是GPU,操作系统是Linux还是别的,Python使用哪个版本,以及想下载哪个版本的Tensorflow来进行选择,得到相应的安装命令。如下图所示。

Tensorflow下载

然后按照图中的命令执行就可以啦~速度要快很多哦。

2.  在Jupyter notebook下使用Keras Tensorflow时遇到的显存不释放、占满等问题

在运行Keras或者Tensorflow时我们会发现个问题,如果你使用的是具有多个GPU的服务器,那么程序会自动调用所有的能够调用到的资源,在每个GPU上挤占一块

评论 1
成就一亿技术人!
拼手气红包6.0元
还能输入1000个字符
 
红包 添加红包
表情包 插入表情
 条评论被折叠 查看
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值