禁用GPU版本TensorFlow,切换到CPU版本TensorFlow。

时间:2024-07-15 23:04:26

#禁用gpu版本TensorFlow,因为CUDA号码从0开始,这里直接让CUDA使用-1的GPU,自然就无法使用gpu了。

代码前面加入:

import os
os.environ["CUDA_VISIBLE_DEVICES"]="-1"    
import tensorflow as tf
Environment Variable Syntax    Results
CUDA_VISIBLE_DEVICES=1    Only device 1 will be seen
CUDA_VISIBLE_DEVICES=0,1    Devices 0 and 1 will be visible
CUDA_VISIBLE_DEVICES="0,1"    Same as above, quotation marks are optional
CUDA_VISIBLE_DEVICES=0,2,3    Devices 0, 2, 3 will be visible; device 1 is masked

如果安装了GPU,默认使用GPU版本的TensorFlow
---------------------
作者:湾区人工智能
来源:****
原文:https://blog.****.net/BTUJACK/article/details/82622966
版权声明:本文为博主原创文章,转载请附上博文链接!