1. win10 安装CUDA、CUDNN
组合1(cuda、cudnn安装成功 tensorflow安装失败):
1.CUDNN6.0下载
2.CUDA8.0下载
3.安装方法:win10下CUDA和CUDNN的安装
组合2(已成功安装):
1.CUDA9.0.176下载
2.CUDNN7.0.5下载
3.tensorflow1.5.0:pip install tensorflow==1.5.0
4.安装方法:win10下CUDA和CUDNN的安装
2. Uuntu16.04 LINUX 安装CUDA CUDNN
组合1:
1.CUDA9.0下载
2.CUDNN linux下载
3.tensorflow1.5.0:pip install tensorflow-gpu==1.5.0
安装方法:Ubuntu16.04下安装cuda和cudnn
3.linux系统下的pytorch的下载安装
PyTorch安装:
命令:conda install pytorch torchvision cudatoolkit=9.0 -c pytorch -c defaults -c numba/label/dev
4.注:
cuda与cudnn的安装必须对应版本(cuda8.0-cudnn6.0)
tensorflow社区
win10下CUDA和CUDNN的安装
记录TensorFlow与Python, CUDA, cuDNN的版本对应关系
Win10下 tensorflow gpu版安装
Ubuntu16.04下安装cuda和cudnn
旧版本的TensorFlow(高版本tensorflow需要安装高版本cuda与cudnn)
对于1.15版本,CPU和GPU支持包含在一个软件包中: pip install --pre "tensorflow==1.15.*"
对于1.14和更早的版本,CPU和GPU软件包是分开的:pip install tensorflow==1.14 # CPU
pip install tensorflow-gpu==1.14 # GPU
python3.7安装tensorflow-gpu:
pip install tensorflow==1.13.0rc1