首页 > 其他 > 详细

检测tensorflow2和pytorch1.x的GPU环境是否配合正确

时间:2021-08-29 21:51:10      阅读:19      评论:0      收藏:0      [点我收藏+]
import tensorflow as tf
print(tf.__version__)
print(‘tf2‘, tf.config.list_physical_devices(‘GPU‘))
print(‘tf2‘, tf.test.is_gpu_available(), tf.__version__)	# 不再推荐

import torch
print(‘torch18‘, torch.cuda.is_available(), torch.__version__)

检测tensorflow2和pytorch1.x的GPU环境是否配合正确

原文:https://www.cnblogs.com/hsiangyu-meng/p/15202724.html

(0)
(0)
   
举报
评论 一句话评论(0
关于我们 - 联系我们 - 留言反馈 - 联系我们:wmxa8@hotmail.com
© 2014 bubuko.com 版权所有
打开技术之扣,分享程序人生!