使用anaconda进行tensorflow的cpu版本和gpu版本的安装步骤

1.cpu版本安装

首先在anaconda中创建一个环境命名为TF2,python版本为3.7,然后按open terminal后,输入:

pip install tensorflow==2.0.0 -i http://pypi.douban.com/simple --trusted-host pypi.douban.com

请稍等,然后测试安装是否成功,然后在环境的命令行中输入:

[En]

Wait a while; then test if the installation is successful, and type in the command line of the environment:

python

后输入:

import tensorflow as tf
tf.__version__

正确显示tensorflow版本则安装成功。

2.gpu版本安装

首先在anaconda中创建一个环境命名为TF2-GPU,python版本为3.7,然后按open terminal后,输入:

conda search cuda

寻找版本号,本人是cuda10.0.130版本,所以这里输入:

conda install cudatoolkit==10.0.130

然后查找对应的版本号并输入:

[En]

Then look for the corresponding version number and enter:

conda search cudnn

本人是cudnn7.6.0版本,所以这里输入:

conda install cudnn==7.6.0

接着进行tensorflow-gpu版本的安装,输入:

pip install tensorflow-gpu==2.0.0 -i http://pypi.douban.com/simple --trusted-host pypi.douban.com

在这里等需要很长时间。安装完成后,测试安装是否成功,然后输入环境的命令行:

[En]

It will take a long time to wait here. After the installation is complete, test whether the installation is successful or not, and type in the command line of the environment:

python

接着输入:

import tensorflow as tf
tf.__version__
tf.test.is_gpu_available()

如果结果显示Ture,则GPU版本安装成功。

注意:tensorflow的cpu版本和gpu版本不要使用一个环境,否则会造成环境错乱。

Original: https://blog.csdn.net/qq_45195178/article/details/123845903
Author: 小王不留行
Title: 使用anaconda进行tensorflow的cpu版本和gpu版本的安装步骤

原创文章受到原创版权保护。转载请注明出处:https://www.johngo689.com/496767/

转载文章受原作者版权保护。转载请注明原作者出处!

(0)

大家都在看

亲爱的 Coder【最近整理,可免费获取】👉 最新必读书单  | 👏 面试题下载  | 🌎 免费的AI知识星球