TensorFlow(一)——conda版安装

1.安装anaconda:

https://repo.continuum.io/archive/index.html选择相应版本,我选择的是最新的版本5.2.0:

wget https://repo.continuum.io/archive/Anaconda3-5.2.0-Linux-x86_64.sh

之后直接进行安装:

bash Anaconda3-5.2.0-Linux-x86_64.sh

之后按照步骤进行就可以,最后需要确认添加路径到.bashrc。(有时候不会自动添加,则需要手动添加)

手动添加:

gedit ~/.bashrc

在最后一行添加:

export PATH=~/anaconda3/bin:$PATH

无论手动还是自动都要重启环境变量:

source ~/.bashrc

就可以了,可以查看python版本是否已经变成anaconda了。

2.安装TensorFlow

参考网站:https://github.com/conda-forge/tensorflow-feedstock

只有CPU版本可以安装:

conda config --add channels conda-forge

conda install tensorflow

这时候进入python,import tensorflow 成功就是安装成功了。

以下是安装tensorflow-gpu的步骤: 1. 安装CUDA Toolkit和cuDNN库 TensorFlow GPU本需要CUDA Toolkit和cuDNN库的支持。请根据您的系统环境下载适合的本。推荐使用CUDA 10.0和cuDNN 7.4.2。下载链接如下: - CUDA Toolkit 10.0:https://developer.nvidia.com/cuda-10.0-download-archive - cuDNN 7.4.2:https://developer.nvidia.com/cudnn 安装CUDA Toolkit和cuDNN库时,请注意设置环境变量,以便TensorFlow能够找到它们。 2. 创建conda虚拟环境 您可以使用conda创建一个虚拟环境来安装和管理TensorFlow GPU本以及其他软件包。打开终端并运行以下命令: ```bash conda create -n tensorflow-gpu python=3.7 ``` 这个命令将创建一个名为“tensorflow-gpu”的conda虚拟环境,其中Python的本是3.7。 3. 激活conda虚拟环境 使用以下命令激活conda虚拟环境: ```bash conda activate tensorflow-gpu ``` 4. 安装TensorFlow GPU本 通过以下命令安装TensorFlow GPU本: ```bash pip install tensorflow-gpu==2.0.0 ``` 这个命令将安装TensorFlow 2.0.0 GPU本。 5. 验证安装 要验证TensorFlow GPU是否正确安装,请使用以下Python代码: ```python import tensorflow as tf print(tf.test.is_gpu_available()) ``` 如果输出的结果为True,则表示TensorFlow GPU安装正确。 提示:安装TensorFlow GPU本之前,请确保您的GPU支持CUDA。您可以在以下链接中查找支持CUDA的GPU:https://developer.nvidia.com/cuda-gpus
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值