tensorflow
James-J
这个作者很懒,什么都没留下…
展开
-
os.environ["CUDA_DEVICE_ORDER"] = "PCI_BUS_ID" os.environ["CUDA_VISIBLE_DEVICES"] = "0"
os.environ[“CUDA_DEVICE_ORDER”] = “PCI_BUS_ID” # 按照PCI_BUS_ID顺序从0开始排列GPU设备 os.environ[“CUDA_VISIBLE_DEVICES”] = “0” #设置当前使用的GPU设备仅为0号设备 设备名称为'/gpu:0' os.environ[“CUDA_VISIBLE_DEVICES”] = “1” #设置当前...原创 2018-10-26 14:48:22 · 41246 阅读 · 5 评论 -
Tensorflow入门:一个比较完整的MNIST数据集训练代码
一个较为完整的mnist数据集训练模型 全连接模型 精确度接近98% 使用指数衰减学习率、滑动平均、正则化、模型保存与读取 过程中需要联网下载mnist数据集 tensorflow的GPU版本运行 并且设置GPU按需分配# -*- coding: utf-8 -*- #import tensorflow as tffrom ...原创 2018-10-26 16:16:56 · 1058 阅读 · 1 评论 -
win10安装tensorflow gpu版本
一、安装cuda查看电脑支持的cuda版本安装cuda(相当于运算平台)才能使得GPU用于我们所需的复杂的计算首先打开控制面板,然后搜索“nvidia 控制面板”,打开来到这个界面,点击帮助中的系统信息点击组件后直接查看cuda版本号,我的是9.1下载安装cuda 然后按照这个版本去下:9.0版本: https://developer.nvidia.co...原创 2019-02-26 18:31:05 · 772 阅读 · 0 评论