在linux系统上使用conda 安装GPU版本TensorFlow-GPU(详细步骤)

文章目录

*

+ 使用conda 还是miniconda
+
*
一、下载miniconda 可以选择python版本等信息:
二、安装miniconda,根据提示按Enter,和输出yes
三、创建虚拟环境
四、激活虚拟环境、
+ 安装Tensorflow
+
*
五、可以先查看版本安装命令
六、安装tensorflow-gpu==2.2.0
七、查看tensorflow版本以及是否可以使用gpu

使用conda 还是miniconda

conda较科学库会比较多,以后不想单独安装要使用的每个包。并且现在安装需要花费一些时间和3G左右的内存空间就可以用conda, miniconda只会安装一些基础都软件包,因此你没有足够的空间或者想要快速都访问Python 和 conda 命令并且希望稍后整理其他程序,就可以下载miniconda。对于安装TensorFlow-GPU,在Linux系统上,并且考虑磁盘空间,以及并不会用到太多都其他库选择我会选择安装miniconda。

下载地址

conda: https://www.anaconda.com/products/distribution

miniconda: https://docs.conda.io/en/latest/miniconda.html

一、下载miniconda 可以选择python版本等信息:

此处 _py38_可看出版本为python3.8

wget --quiet  https://repo.anaconda.com/miniconda/Miniconda3-py38_4.12.0-Linux-x86_64.sh

在linux系统上使用conda 安装GPU版本TensorFlow-GPU(详细步骤)
二、安装miniconda,根据提示按 Enter ,和输出 yes
bash Miniconda3-py38_4.12.0-Linux-x86_64.sh

如图:

在linux系统上使用conda 安装GPU版本TensorFlow-GPU(详细步骤)
在linux系统上使用conda 安装GPU版本TensorFlow-GPU(详细步骤)

这里有个提示意思是下次你进去服务器会自动启动conda(base)环境不想这样都话可以使用下面命令:

conda config --set auto_activate_base false

重启终端,测试conda -V ,安装成功

在linux系统上使用conda 安装GPU版本TensorFlow-GPU(详细步骤)
conda镜像

conda config --show channels

conda config --remove-key channels

conda config --remove channels https://xxx

conda config --add channels http://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/msys2/
conda config --add channels http://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge/
conda config --add channels http://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --add channels http://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/

conda config --set show_channel_urls yes
三、创建虚拟环境
conda create -n my-env python=3.8
四、激活虚拟环境、
conda activate my-env

#有时候第一不行试试这个
source activate my-env

退出虚拟环境

conda deactivate

安装Tensorflow

注:conda有的版本自动安装CUDA/cuDNN,若果没有安装自行安装CUDA/cuDNN,方法在后面

五、可以先查看版本安装命令
conda search tensorflow-gpu

在linux系统上使用conda 安装GPU版本TensorFlow-GPU(详细步骤)
六、安装tensorflow-gpu==2.2.0
conda install tensorflow-gpu==2.2.0

在linux系统上使用conda 安装GPU版本TensorFlow-GPU(详细步骤)
七、查看tensorflow版本以及是否可以使用gpu

需要先进去虚拟环境,再输入python,

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

在linux系统上使用conda 安装GPU版本TensorFlow-GPU(详细步骤)

安装CUDA/cuDNN

使用下面命令查看conda可安装cudatoolkit版本信息

conda search cudatoolkit

使用下面命令进行安装CUDN 、cuDNN一样可以查看版本信息

conda install cudatoolkit==xx.xx
conda install cudnn==xx.xx

Original: https://blog.csdn.net/Peyzhang/article/details/125555150
Author: Peyzhang
Title: 在linux系统上使用conda 安装GPU版本TensorFlow-GPU(详细步骤)

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

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

(0)

大家都在看

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