tensorflow使用GPU小记
指定使用的GPU:
import os
os.envrion["CUDA_VISIBLE_DEVICES"]="0,1"
查看GPU使用情况
在cmd中输入:nvidia-smi 要想不断刷新记录:nvidia-smi -l
出现的问题
报错:
failed to create cublas handle:CUBLAS_STATUS_ALLOC_FAILED
原因是显存不足,例如同时开启了jupyter notebook 和pycharm,或者pycharm同时运行了多个需要GPU的程序,全部关掉保留一个就好了。
tensorboard启动
cmd中输入 tensorboard --logdir="logs"
logs是自己保存日志起的文件名