在linux系统上使用conda 安装GPU版本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

在这里插入图片描述

二、安装miniconda,根据提示按Enter,和输出yes
bash Miniconda3-py38_4.12.0-Linux-x86_64.sh

如图:


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

conda config --set auto_activate_base false

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

在这里插入图片描述
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

在这里插入图片描述

六、安装tensorflow-gpu==2.2.0
conda install tensorflow-gpu==2.2.0

在这里插入图片描述

七、查看tensorflow版本以及是否可以使用gpu

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

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

[外链图片转存失败,源站可能有防盗链机制,建议将图片保存下来直接上传(img-B7m0X1Iv-1656645097728)(C:\Users\peyzhang\AppData\Roaming\Typora\typora-user-images\image-20220701110450594.png)]

安装CUDA/cuDNN

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

conda search cudatoolkit

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

conda install cudatoolkit==xx.xx
conda install cudnn==xx.xx
### 安装 TensorFlow-GPU 的方法 为了在 Linux 服务器上使用 Anaconda 正确配置并安装 TensorFlow-GPU,可以通过以下几种方式进行操作: #### 方法一:通过 Conda 命令直接安装 可以直接利用 `conda` 来安装指定版本TensorFlow-GPU。例如,要安装 TensorFlow-GPU 版本 1.12.0,可以运行以下命令: ```bash conda install tensorflow-gpu==1.12.0 ``` 此方法适用于 Conda 已经支持的 TensorFlow-GPU 版本[^1]。 #### 方法二:通过 Pip 安装特定版本 如果目标 TensorFlow-GPU 版本未被 Conda 支持,则可以在激活的 Anaconda 虚拟环境中使用 `pip` 进行安装。例如,对于 TensorFlow-GPU 1.13.2 可以执行以下命令: ```bash pip install tensorflow-gpu==1.13.2 ``` 这种方法允许更灵活地选择不同版本TensorFlow-GPU[^2]。 #### 方法三:离线安装 TensorFlow-GPU 当网络条件受限时,可以选择下载对应版本的 `.whl` 文件并通过本地文件路径完成安装。例如,在 Python 3.7 环境下安装 TensorFlow-GPU 2.3.0,可先获取对应的轮子文件(wheel file),然后运行以下命令: ```bash pip install tensorflow_gpu-2.3.0-cp37-cp37m-manylinux2010_x86_64.whl ``` 该过程需确保所选 wheel 文件与系统的架构以及 Python 解释器版本相匹配[^3]。 #### 额外提示:混合工具链安装其他依赖项 有时可能还需要额外安装一些库来满足项目需求。在这种情况下,即使某些包无法通过 Conda 获取,也可以借助内置的 Pip 实现安装目的。比如安装 NumPy 库至当前活动的虚拟环境里,可以用下面这条指令实现: ```bash python -m pip install numpy ``` 这一步骤展示了如何结合两种工具的优势来构建完整的开发环境[^4]。
评论 2
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值