202103-17 解决PyTorch与CUDA版本不匹配的问题 1.CUDA驱动和CUDAToolkit对应版本表一:CUDA驱动及CUDAToolkit最高对应版本最新可查阅官方文档注:驱动是向下兼容的,其决定了可安装的CUDAToolkit的最高版本。2.CUDAToolkit版本及其可用PyTorch对应版本(参考官网)表二:CUDAToolkit版本及可用PyTorch对应关系CUDAToolkit版本可用PyTorch版本7.50.4.1,0.3.0,0.2.0,0.1.12-0.1.68.01.1.0,1.0.0,0.4.... 继续阅读 >
202010-08 解决tensorflow/keras时出现数组维度不匹配问题 一、环境Ubuntu16.04tensorflow1.4.0keras2.1.3二、训练数据时报错:ValueError:Errorwhencheckingtarget:expectedmodel_2tohaveshape(None,3)butgotarraywithshape(4,1原因:数组的维度不正确。三、解决思路一开始的想法:tensorflow和keras的版本不兼容?经过多次安装不同版本的tensorflow和keras,发现问题依旧存在。而且在python环境下使用命令:importtensorflow或者importkeras时,报错:/home... 继续阅读 >