在开始本教程之前,请验证LD_LIBRARY环境变量或ldconfig包含$CUDA_ROOT/extras/CUPTI/lib64,其中包含CUDA分析工具界面(CUPTI)的库。
允许GPU内存增长 默认情况下,TensorFlow将几乎所有GPU(归属于CUDA_VISIBLE_DEVICES)的所有GPU内存映射到进程。
练习:模型体系结构inference()与cuda-convnet中指定的CIFAR-10模型略有不同。特别是,Alex的原始模型的顶层是局部连接的并且没有完全连接。

扫码关注腾讯云开发者
领取腾讯云代金券
Copyright © 2013 - 2026 Tencent Cloud. All Rights Reserved. 腾讯云 版权所有
深圳市腾讯计算机系统有限公司 ICP备案/许可证号:粤B2-20090059
粤公网安备44030502008569号
腾讯云计算(北京)有限责任公司 京ICP证150476号 | 京ICP备11018762号
