Tensorflow中使用CPU及GPU的参数设置
Tensorflow中使用CPU及GPU的参数设置在进行SVM机器学习训练时,报dst Tensorflow is not initialized错误,为显存使用溢出错误。解决方式为使用CPU或者限制GPU显存。1、设置使用GPU还是CPU:import osos.environ['CUDA_VISIBLE_DEVICES'] = "0" 2、安装GPU版本Tensorflow,只有一...
原创
2019-03-05 20:30:29 ·
3402 阅读 ·
0 评论