参考文章keras & tensorflow 列出可用GPU 和 切换CPU & GPU
可能sklearn里面的一堆自带文本分类算法,并不能用到GPU???😓反正没有实验成功
安装:
pip/conda install tensorflow-gpu==1.14.0
pip/conda install keras
验证:
from tensorflow.python.client import device_lib
print(device_lib.list_local_devices())
from keras import backend as K
K.tensorflow_backend._get_available_gpus()
使用:
import os
os.environ["CUDA_DEVICE_ORDER"]="PCI_BUS_ID"
os.environ["CUDA_VISIBLE_DEVICES"] = "0" GPUid
os.environ["CUDA_VISIBLE_DEVICES"] = "-1" 为使用CPU
网友评论