Tensorflow-gpu的安装+pycharm

导读:本篇文章讲解 Tensorflow-gpu的安装+pycharm,希望对大家有帮助,欢迎收藏,转发!站点地址:www.bmabk.com


一、基本步骤

1.安装Anaconda

2.在Anaconda中创建环境

3.安装tensorflow

4.在pycharm中配置环境

二、开始安装

1.安装Anaconda

1.可以去https://www.anaconda.com/distribution/(Anaconda官网)下载对应的Anaconda
2.在安装完之后添加Anaconda进电脑的环境变量
具体方法 : 我的电脑–属性–高级系统设置–环境变量–用户变量–path–添加Anconada在这里插入图片描述

2.在Anaconda中创建环境

在安装下载完之后,进入Anaconda Prompt
(具体方法:开始–Anaconda文件夹下)

输入conda info --envs 检查Anaconda环境,
一般是只有一个base环境
检查目前有哪些版本的python可以安装:

 conda search --full-name python 

选择一个python的版本
然后创建环境;
输入

conda create -n 环境名字(随便取) python=3.7

在这里取一个tensorflow

然后输入

conda info --envs

查看新环境是否创建成功

在创建成功之后进入创建的好的环境
输入 activate tensorflow
前面就会有出现一个括号(tensorflow)在最前面
这就表明了进入了环境

安装tensorflow

接着在新建好的环境里面输入

pip install tensorflow-gpu

conda install tensorflow-gpu=2.0.0

如果速度慢的话也可以用镜像源安装

python -m pip install -i https://pypi.tuna.tsinghua.edu.cn/simple tensorflow-gpu==1.3.0

在安装完之后输入
pip list 检查是否安装上

4.在pycharm中配置环境

打开pycharm
在这里插入图片描述
选择add
在这里插入图片描述
选取Anaconda3—-envs—tensoflow(自己取的环境变量名)—python.exe
接着Conda Envirment中,也是在红框位置选取:Anaconda3—-envs—tensoflow(自己取的环境变量名)—python.exe就完成了在pycharm中的配置
在这里插入图片描述
在pycharm中建立文件输入

import tensorflow as tf

version = tf.__version__
gpu_ok = tf.test.is_gpu_available()

print("tf version:",version,"\nuse GPU",gpu_ok)

验证是否安装成功
若成功则输出

tf version: 2.0.0 
use GPU True

(第一次时间程序运行时间可能很长,请耐心等待)


版权声明:本文内容由互联网用户自发贡献,该文观点仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 举报,一经查实,本站将立刻删除。

文章由极客之音整理,本文链接:https://www.bmabk.com/index.php/post/66903.html

(0)
小半的头像小半

相关推荐

极客之音——专业性很强的中文编程技术网站,欢迎收藏到浏览器,订阅我们!