1、tensorflow中的cuda版本问题
CUDA有两种版本,一种是驱动版本 driver version,查看命令:nvidia-smi(右上角);一种是运行时版本runtime version,查看命令:pip list 或conda list(包管理器+list)cudatoolkit便是。
两种版本必须对应,对应关系如下:
CUDA 9.1 387.xx
CUDA 9.0 384.xx
CUDA 8.0 375.xx (GA2)
CUDA 8.0 367.4x
CUDA 7.5 352.xx
CUDA 7.0 346.xx
CUDA 6.5 340.xx
CUDA 6.0 331.xx
CUDA 5.5 319.xx
CUDA 5.0 304.xx
CUDA 4.2 295.41
CUDA 4.1 285.05.33
CUDA 4.0 270.41.19
CUDA 3.2 260.19.26
CUDA 3.1 256.40
CUDA 3.0 195.36.15
如果不一致,一般来说需要删除runtime version 重新安装来适应当前drive version。
流程:
删除cudnn->删除cudatoolkit->安装cudatoolkit->安装cudn->安装tensorflow-gpu
2、windows从本地上传到github文件