我是靠谱客的博主 乐观鸡,最近开发中收集的这篇文章主要介绍TensorFlow 安装1 Windows系统下安装2. Ubuntu下安装tensorflow3. 删除4. 测试GPU与CPU版本5. 对环境处理6. 本文参考,觉得挺不错的,现在分享给大家,希望可以做个参考。

概述

1 Windows系统下安装

 

1.1 安装conda

(anaconda即可,一般anaconda安装位置:C:ProgramDataAnaconda3)

 

  • 查看conda下的安装情况

conda info --envs

可能出现连接网络的问题,切换为无线

 

  • 为通过pip安装,首先需要更新pip

python -m pip install --upgrade pip

有必要使用 pip install --upgrade setuptools

 

  • 创建Tensorflow环境

conda create -n tensorflow-gpu python=3.6

 

1.2 安装

  • 默认CPU版本

pip install --upgrade --ignore-installed tensorflow

  • 安装GPU版本

GPU 版本安装需要先安装cuda + CuDNN

CUDA下载9.0稳定版本exe文件

cudnn的3个文件夹拷贝至cuda的安装目录下

pip install --upgrade --ignore-installed tensorflow-gpu==1.12

 

注意:tfgpu1.13版本仅支持cuda10

技巧:更换conda下载源:(pypi就是python仓库)

(临时使用)pip install -i https://pypi.tuna.tsinghua.edu.cn/simple some-package

(设为默认)pip install pip -U

     pip config set global.index-url https://pypi.tuna.tsinghua.edu.cn/simple

  • 测试是否安装成功

python

import tensorflow as tf

  • 进入环境

activate tensorflow-gpu

  • 退出环境

deactivate

 

2. Ubuntu下安装tensorflow

同样首先安装anaconda,系统显卡设为独显,对应版本的CUDA与cudnn

下载的文件是:cuda_9.0.176_384.81_linux.run

cudnn-9.0-linux-x64-v7.1.tgz

  • 命令:

sudo sh cuda_9.0.176_384.81_linux.run -override

注意不要安装默认驱动

 

  • 打开文件

 sudo gedit ~/.bashrc

添加

export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda-9.0/lib64

export PATH=$PATH:/usr/local/cuda-9.0/bin

export CUDA_HOME=$CUDA_HOME:/usr/local/cuda-9.0

 

  • 测试

nvcc -V

 

  • 安装cuDNN

tar -xzvf cudnn-8.0-linux-x64-v6.0.tgz

sudo cp cuda/include/cudnn.h /usr/local/cuda/include

sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64

sudo chmod a+r /usr/local/cuda/include/cudnn.h /usr/local/cuda/lib64/libcudnn*

 

3. 删除

CPU:

激活tensorflow:activate tensorflow

输入:pip uninstall tensorflow

Proceed(y/n):y

GPU:

激活tensorflow:activate tensorflow-gpu

输入:pip uninstall tensorflow-gpu

Proceed(y/n):y

删除conda下的环境:

conda remove -n 环境名 --all

退出conda下的环境:

deactivate 环境名

 

4. 测试GPUCPU版本

 

import os

from tensorflow.python.client import device_lib

os.environ["TF_CPP_MIN_LOG_LEVEL"] = "99"

 

if __name__ == "__main__":

    print(device_lib.list_local_devices())

 

5. 对环境处理

创建新环境

conda create -n rcnn python=3.6

删除环境

conda remove -n rcnn --all

重命名环境

conda 其实没有重命名指令,实现重命名是通过 clone 完成的,分两步:

  • 先 clone 一份 new name 的环境
  • 删除 old name 的环境

比如,想把环境 rcnn 重命名成 tf

第1步

conda create -n tf --clone rcnn

Source:      /anaconda3/envs/rcnn
Destination: /anaconda3/envs/tf
Packages: 37
Files: 8463

第2步

conda remove -n rcnn --all

 

6. 本文参考

出现问题解决:https://blog.csdn.net/qq_23116521/article/details/80929082

清华镜像帮助网站: https://mirrors.tuna.tsinghua.edu.cn/help/pypi/

anaconda无法使用activate激活环境   https://blog.csdn.net/kdongyi/article/details/81905494

下载源:(阿里)http://mirrors.aliyun.com/pypi/simple/   (使用需要在后加  --trusted-host mirrors.aliyun.com)

     (清华)https://pypi.tuna.tsinghua.edu.cn/simple

 

 

最后

以上就是乐观鸡为你收集整理的TensorFlow 安装1 Windows系统下安装2. Ubuntu下安装tensorflow3. 删除4. 测试GPU与CPU版本5. 对环境处理6. 本文参考的全部内容,希望文章能够帮你解决TensorFlow 安装1 Windows系统下安装2. Ubuntu下安装tensorflow3. 删除4. 测试GPU与CPU版本5. 对环境处理6. 本文参考所遇到的程序开发问题。

如果觉得靠谱客网站的内容还不错,欢迎将靠谱客网站推荐给程序员好友。

本图文内容来源于网友提供,作为学习参考使用,或来自网络收集整理,版权属于原作者所有。
点赞(53)

评论列表共有 0 条评论

立即
投稿
返回
顶部