Linux快速搭建tensorflow-gpu 1.15.0环境

1.创建虚拟环境:tmp_python

conda create -n tf python=3.6

温馨提醒:python3.8暂不支持tensorflow-gpu 1.15版本,请使用python3.6版本

2.显示自己创建的虚拟环境

conda env list

3.进入创建虚拟环境

source activate tf

4.安装指定 版本的TensorFlow

pip3 install tensorflow-gpu==1.15.0

5.安装的包(务必执行)

conda install cudatoolkit=10.0 -c https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/linux-64/
conda install cudnn=7.6.5 -c https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/linux-64/

小技巧:若安装其他版本,先查看Tensorflow-gpu对应支持的cuda版本,然后进入清华镜像网站,搜索cudnn,然后找到对应的cuda版本

6.验证是否安装成功

先输入python进入交互式界面,然后用如下命令验证是否gpu版本启用

import tensorflow as tf
tf.test.is_gpu_available()

若最后显示True,则恭喜你,安装成功!
在这里插入图片描述

已标记关键词 清除标记
相关推荐
©️2020 CSDN 皮肤主题: 大白 设计师:CSDN官方博客 返回首页