torch tensorflow测试GPU配置是否成功以及相关问题
目录torch问题tensorflow-gpu问题参考torchimport torch'''配置成功会输出一个device ID:0未配置成功会输出False'''torch.cuda.current_device()'''或者,返回True or False'''print(torch.cuda.is_available())问题Torch not compiled with CUDA enabled:pytorch对应的cudatoolkit版本不对tensorflo
原创
2021-11-15 19:31:20 ·
1081 阅读 ·
0 评论