GPU基础
张金玉
这个作者很懒,什么都没留下…
展开
-
kubernetes docker GPU pod启动
apiVersion: v1kind: Podmetadata: name: cuda-vector-addspec: restartPolicy: OnFailure containers: - name: cuda-vector-add # https://github.com/kubernetes/kubernetes/blob/v1.7.11/tes...原创 2018-12-27 13:57:12 · 956 阅读 · 0 评论 -
nvidia命令入门
一操作系统基础信息如果操作系统不支持lspci命令,需要安装root@b57334c4f529:~# apt-get install pciutils -y #或者centos yum install pciutils -y1.1 首先验证你是否有nvidia的显卡(http://developer.nvidia.com/cuda-gpus这个网站查看你是否有支持gpu的显卡)roo...原创 2018-12-11 11:10:03 · 6811 阅读 · 0 评论 -
GPU单机单卡/单机多卡/多机单卡/多机多卡的实现过程
一单机单卡# -*- coding: UTF-8 -*-#单机单卡#对于单机单卡,可以吧参数和计算都定义再GPU上,不过如果参数模型比较大,显存不足等,就得放在CPU上import tensorflow as tfwith tf.device("/cpu:0"): #也可以放在GPU上 w = tf.get_variable('w', (2,2), tf.float32, i...原创 2019-07-20 18:54:48 · 7909 阅读 · 1 评论