Ubuntu下安装Anaconda3、tensorflow-gpu

人工智能107

Ubuntu下安装Anaconda3、tensorflow-gpu(RTX3080Ti)

一、安装Anaconda

从官网下载[Linux版本的Anaconda3],选择自己想要下载的版本即可(https://www.anaconda.com/products/distribution#Downloads)
Ubuntu下安装Anaconda3、tensorflow-gpu
下载完成后通过指令"bash 文件路径/Anaconda3-2021.11-Linux-x86_64.sh"就可以开始安装了,安装过程中,需要输入两个"yes",第一个是是否接受许可条款,输入yes
Ubuntu下安装Anaconda3、tensorflow-gpu
然后会询问是否要修改安装路径,不修改的话,直接enter就开始安装
Ubuntu下安装Anaconda3、tensorflow-gpu
安装好需要的一些包之后,会询问是否要初始化环境,输入yes,就不需要自己再去加环境变量了
Ubuntu下安装Anaconda3、tensorflow-gpu
如果不小心输成no了,也可以自己vim ~/.bashrc 在最后一行加入环境变量即可
export PATH=$PATH:【你的安装目录】
最后保存更改:source ~/.bashrc
出现下面这个界面就说明安装成功了
Ubuntu下安装Anaconda3、tensorflow-gpu
切换到Anaconda3安装目录下的bin,用source activate base激活conda,不然直接使用conda命令,会提示"未找到命令"。这一步之后就可以开始你的任何操作了!

; 二、创建虚拟环境,并安装torch

  1. 创建虚拟环境
    conda create -n envname python=3.X
    创建好之后,也会给出激活和退出的命令
    Ubuntu下安装Anaconda3、tensorflow-gpu
    使用 conda activate envname进入刚刚创建的虚拟环境,安装的tensorflow、torch等包就安装到了自己的环境中了。
  2. 安装torch
    进入pytorch官网所有版本的下载地址,选择一个符合自己GPU算力的版本,复制命令到命令行中即可
    Ubuntu下安装Anaconda3、tensorflow-gpu
    使用上面的命令安装,可以直接把pytorch和cuda都装好,所以如果pytorch和tensorflow都经常用的朋友,现状pytorch可以更省事哦。
  3. CUDA算力查看
    在本机中安装了cuda,则切换到如下目录中去查看对应的cuda驱动版本和算力大小,RTX3080Ti支持的驱动版本是11.3,对应算力为8.6,所以在安装CUDA时需要安装支持算力为8-8.6的版本。10.1是当前运行的CUDA版本,之前安装到本机上的。
    也可直接用命令"nvidia-smi"查看
    Ubuntu下安装Anaconda3、tensorflow-gpu
    RTX3080Ti以上系列的都要装CUDA11.0以上,如果装10.X,在代码运行过程中就不能使用GPU。
  4. 测试是否安装成功
    进入python环境,输入一下命令
import torch
torch.zeros(1).cuda()

如果输出相应的tensor张量,及设备是CUDA就成功了
Ubuntu下安装Anaconda3、tensorflow-gpu
如果报错是GeForce RTX 3080 with CUDA capability sm_86 is not compatible with the current PyTorch installation. The current PyTorch install supports CUDA capabilities sm_37 sm_50 sm_60 sm_61 sm_70 sm_75 compute_37.那么一定是cuda版本的问题了。安装cuda版本不对。

三、安装tensorflow-gpu

  1. 配置cuda和cudnn环境
    安装tensorflow-gpu之前要安装对应的cuda和cudnn,在安装torch包的时候,已经安装好了cuda,在这里只需要选择与上面cuda相应的cudnn就可以了。
    用命令"conda search cudnn -c conda-forge"可查看CUDA支持的cudnn版本,用conda命令安装一下即可
    Ubuntu下安装Anaconda3、tensorflow-gpu
    我们安装与11.3对应的cudnn8.2.1
    Ubuntu下安装Anaconda3、tensorflow-gpu
  2. 安装tensorflow-gpu
    cuda和cudnn都已经安装好了,直接命令行中安装gpu版的tensorflow即可。
    安装命令: "conda install tensorflow-gpu2.6.0",如果出错,用pip安装即可
    "pip install tensorflow-gpu2.6.0"
    下面的环境在在GPU为RTX3080Ti中是亲测可用的
    cuda ==11.0或11.1
    cudnn ==8.2.1
    python ==3.6/3.7
    TensorFlow-gpu ==2.5.0
    Pytorch ==1.7.1

在RTX3090Ti中,各版本稍微不一样,需要注意的一点是,tensorflow-gpu要安装2.X,安装1.X一样会出错。

        cuda ==11.3
        cudnn ==8.2.1
        python ==3.9
        TensorFlow-gpu ==2.6.0
        Pytorch ==1.10.1
    ```

Original: https://blog.csdn.net/weixin_43868930/article/details/118902193
Author: camellia_meizi
Title: Ubuntu下安装Anaconda3、tensorflow-gpu