© 1999-2048 dssz.net 粤ICP备11031372号
[其它] 基于tensorflow指定GPU运行及GPU资源分配的几种方式小结
说明:1. 在终端执行时设置使用哪些GPU(两种方式) (1) 如下(export 语句执行一次就行了,以后再运行代码不用执行) (2) 如下 2. 代码中指定(两种方式) (1) import os os.environ["CUDA_VISIBLE_DEVICES"] = "1" (2) # Creates a graph. with tf.device('/gpu:1'): a = tf.constant([1.0, 2.0, 3.0, 4.0, 5.0, 6.0], shape=[2,<weixin_38691220> 上传 | 大小:49kb