系统版本:ubuntu 18.04
系统架构:x86_64
nvidia driver 驱动版本:440.59
CUDA版本:10.0
cuDNN:7.4.1
gcc:4.8.5
tensorflow-gpu:1.14.0
前提
- step1:确定是否安装了驱动:
nvidia-smi
- step2:查看gcc版本:
gcc --version
若gcc版本太高就更改版本:sudo apt-get install gcc-4.8 g++-4.8
切换gcc版本:sudo update-alternatives --install /usr/bin/gcc gcc /usr/bin/gcc-4.8 100
激活生效:sudo update-alternatives --config gcc
- step3:确定GPU卡可以进行CUDA计算,并且计算力>3.5
到nvidia官网查
- step4:找好对应的版本:
tensorflow版本<—>CUDA版本<—>cuDNN版本。tensorflow,CUDA,cuDNN之间有严格依赖关系,对应版本一定要对应好。
驱动版本与CUDA有关联,CUDA要求驱动大于某个版本。
1.anaconda安装
- step1:下载anaconda安装包
wget https://repo