分类目录:深度学习

以下是分类 深度学习 下的所有文章

Python os.listdir 排序不正确导致数据错误

前段时间在训练模型时,发现了一个莫名的问题,在两台机器上训练结果完全不同,一台机器可以拟合的比较好,另一台完全无法拟合数据。 同时,两台机器拿相同模型weights相同数据进行evaluate时,结果也相差很大。 检查各种问题,发现是读取数据的时候使用Python os.listdir获取路径下的所有文件,而这个方法的返回值是路径下所有文……

conda/anaconda一键安装CUDA cuDNN TensorFlow

按照Nvidia官方安装Cuda 按照官方文档安装CUDA配合对应版本的cuDNN会非常繁琐,如果某些地方出错会导致一系列的错误。 利用conda安装tensorflow会自动安装对应版本的CUDA和cuDNN 直接使用conda install tensorflow一键安装,开箱即用。建议在安装之前先创建虚拟环境。 需要注意的是,目前c……