def is_gpu():
packages = [str(i) for i in pip.get_installed_distributions()]
for item in packages:
if "tensorflow-gpu" in item:
return True
return False
微信
公众号
扫码关注公众号