Ubuntu16.04安装tensorflow-GPU记录

傷城~ 2023-10-06 13:48 45阅读 0赞

## 1.创建conda环境 ##

conda create -n tf-GPU python==3.7

## 2.激活环境 ##

conda activate tf-GPU

## 3.安装tensorflow ##

### 1.首先检查自己的cuda和cudnn的版本号，然后在官网进行对比，找到自己的版本 ###

[tensoerflow官网对应版本][tensoerflow]

使用下面的命令可以看到自己的版本号

cat /usr/local/cuda/version.txt
    
    cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2

![20200729142316282.png][]

我这里是CUDA10.1和CUDNN 7，因此对应的版本号为如下所示：

![20200729142536208.png][]

找到自己的版本号后使用指令进行安装：

pip install tensorflow-gpu==2.1.0

## 4.测试是否安装成功 ##

tf.__version__

![20200729143203601.png][]

也可以使用以下指令测试

import tensorflow as tf
    tf.test.is_gpu_available()

![20200729143335271.png][]

最后为True就说明成功了。

## 遇到问题： ##

1.卸载之前版本强制重装

pip install --upgrade --force-reinstall tensorflow-gpu==1.13.1

[tensoerflow]: https://tensorflow.google.cn/install/source#tested_build_configurations
[20200729142316282.png]: https://img-blog.csdnimg.cn/20200729142316282.png
[20200729142536208.png]: https://img-blog.csdnimg.cn/20200729142536208.png
[20200729143203601.png]: https://img-blog.csdnimg.cn/20200729143203601.png
[20200729143335271.png]: https://img-blog.csdnimg.cn/20200729143335271.png

发表评论取消回复

表情：

评论列表（有 0 条评论，45人围观）

还没有评论，来说两句吧...

相关阅读

相关 Ubuntu安装过程记录

找到Rufus-3.20p.exe版本对于我的Acer AMD a6-4400m电脑 win7 64位系统版本高支持不了版本低制作的启动盘BIOS识别不...

逃离我推掉我的手/ 2024年04月27日 07:54/ 0 赞/ 119 阅读

相关 Xmanager远程访问Ubuntu1604LTS

Xmanager远程访问Ubuntu1604LTS > 简述： > ubuntu1604：xmanager+xdmcp+lightdm+xubuntu(xfce)桌面

喜欢ヅ旅行/ 2022年11月27日 03:09/ 0 赞/ 698 阅读

相关 ubuntu 1604安装php7.4 并安装swoole

安装php7.4 参考 [https://www.mf8.biz/ubuntu-debian-install-php7-3/][https_www.mf8.biz_ubunt

我会带着你远行/ 2022年11月12日 09:56/ 0 赞/ 436 阅读

相关 Ubuntu下Caffe安装记录

总体过程 [Ubuntu16.04 Caffe 安装步骤记录（超详尽）][Ubuntu16.04 Caffe] 手动安装Nvidia显卡驱动，安装Cuda时只

亦凉/ 2022年06月01日 03:29/ 0 赞/ 292 阅读

相关 ubuntu 安装 tensorflow过程记录

1，看python版本： python --version Python 2.7.6 2，因为安装需要python3的版本，切换python版本指定默认

古城微笑少年丶/ 2022年02月02日 11:21/ 0 赞/ 373 阅读

相关 ubuntu server deploy增加对ubuntu 1604支持

[2019独角兽企业重金招聘Python工程师标准>>> ][2019_Python_] ![hot3.png][] 其实ubuntu 14.04 lts的支持时间到2019

淡淡的烟草味﹌/ 2022年01月14日 14:27/ 0 赞/ 424 阅读

相关【日常记录】Ubuntu安装Tomcat

1、打开 [https://tomcat.apache.org/download-80.cgi][https_tomcat.apache.org_download-80.c

曾经终败给现在/ 2022年01月11日 03:39/ 0 赞/ 318 阅读

相关【日常记录】Ubuntu安装jdk

1、首先检查Ubuntu版本 \` root@owen:~ uname -a Linux owen 4.4.0-105-generic 128-Ubunt

Myth丶恋晨/ 2022年01月11日 03:35/ 0 赞/ 335 阅读

相关 ubuntu1604安装mysql

安装mysql非常简单只需要几条命令就可以完成。　　1. sudo apt-get install mysql-server 2. sudo apt-get insta

Dear 丶/ 2021年09月01日 04:57/ 0 赞/ 628 阅读

相关 ubuntu安装经常使用记录

1、将安装文件安装到根目录下最后需要敲下这个命令 sudo make install 2、编译上一层为 cmake .. ,后 make (-j4) ，运行 ./ 二...

朱雀/ 2021年04月01日 19:36/ 0 赞/ 565 阅读