cpu上一个epoch速度是之前的100倍,发现更新环境之后没有用gpu跑
watch -n 1 nvidia-smi查看gpu利用率
conda list查看已安装包,发现没有tensorflow-gpu
pip install tensorflow-gpu
就可以在gpu上跑了
cpu上一个epoch速度是之前的100倍,发现更新环境之后没有用gpu跑
watch -n 1 nvidia-smi查看gpu利用率
conda list查看已安装包,发现没有tensorflow-gpu
pip install tensorflow-gpu
就可以在gpu上跑了