Linux 安装TensorFlow-gpu=1.15版本命令行版
转载自:Linux快速搭建tensorflow-gpu 1.15.0环境
0.查看Python、CUDA、CUDNN对应版本
https://blog.csdn.net/zh515858237/article/details/115708349
NVIDIA官网对应列表
1.创建虚拟环境
% 尽量安装3.6版本python
conda create -n tf python=3.6
2.激活虚拟环境
source activate tf
3.pip安装TensorFlow-gpu
%TensorFlow版本请按照CUDA、CUDNN决定
pip3 install tensorflow-gpu==1.15.0 (pip install也可)
4.命令行安装CUDA和CUDNN
%CUDA、CUDNN版本请按照本机显卡算力决定
% 使用TensorFlow1.15版请安装10.0版以上CUDA及7.4版以上CUDNN
conda install cudatoolkit=10.0 -c https://mirrors.aliyun.com/anaconda/pkgs/main/linux-64
conda install cudnn=7.6.5 -c https://mirrors.aliyun.com/anaconda/pkgs/main/linux-64
5.验证是否安装成功
先输入python进入交互式界面,然后用如下命令验证是否gpu版本启用
import tensorflow as tf
tf.test.is_gpu_available()
若最后显示True,则恭喜你,安装成功!