在Linux服务器基于anaconda安装tensorflow-gpu,超简单详细,亲测有效!

第一步 GPU驱动安装

如果测试运行nvidia-smi,有表格输出说明GPU驱动已安装,可跳过该步骤。

  1. 查看linux的版本以及GPU信息

    (base) zhuhang@JUPITER:~$ cat /proc/version
    Linux version 5.3.0-40-generic (buildd@lcy01-amd64-024) (gcc version 7.4.0 (Ubuntu 7.4.0-1ubuntu1~18.04.1)) #32~18.04.1-Ubuntu SMP Mon Feb 3 14:05:59 UTC 2020
    
    (base) zhuhang@JUPITER:~$ lspci | grep -i nvidia
    02:00.0 VGA compatible controller: NVIDIA Corporation GP102 [GeForce GTX 1080 Ti] (rev a1)
    02:00.1 Audio device: NVIDIA Corporation GP102 HDMI Audio Controller (rev a1)
    82:00.0 VGA compatible controller: NVIDIA Corporation GP104 [GeForce GTX 1080] (rev a1)
    82:00.1 Audio device: NVIDIA Corporation GP104 High Definition Audio Controller (rev a1)
    
  2. 根据系统版本及GPU型号寻找合适的显卡驱动并下载安装

    NVDIA显卡驱动下载地址

    wget -c https://us.download.nvidia.com/XFree86/Linux-x86_64/450.66/NVIDIA-Linux-x86_64-450.66.run
    

    测试:输入nvidia-smi,如有表格输出说明驱动安装成功。

第二步 anaconda 安装

  1. 寻找anaconda的下载链接,

    官方下载地址清华大学镜像

  2. 下载

    wget https://mirrors.tuna.tsinghua.edu.cn/anaconda/archive/Anaconda3-2020.07-Linux-x86_64.sh
    
  3. 安装

    bash Anaconda3-2020.07-Linux-x86_64.sh
    
  4. 更新环境变量source ~/.bashrc

  5. 验证安装

    (base) zhuhang@JUPITER:~$ python3
    Python 3.8.3 (default, Jul  2 2020, 16:21:59)
    [GCC 7.3.0] :: Anaconda, Inc. on linux
    Type "help", "copyright", "credits" or "license" for more information.
    

第三步 建立tensorflow_gpu环境

建立环境,如下命令中py36tensor1为虚拟环境的名称(建议包括python版本和使用框架的名称),python=3.6为指定虚拟环境的python版本。

conda create -n py36tensor1 python=3.6

有关环境的常用命令

conda env remove -n [环境名称]  //删除环境
conda env list   //查看所有环境
conda activate [环境名称] // 激活环境
conda deactivate // 关闭环境 

第四步 安装tensorflow

  1. 搜索指定包
conda search tensorflow-gpu
  1. 激活py36tensor1环境,并安装tensorflow-gpu指定版本
conda activate py36tensor1
conda install tensorflow-gpu=1.13.1
  1. 验证tensorflow安装

    python -c "import tensorflow as tf;print(tf.reduce_sum(tf.random.normal([1000, 1000])))"
    

    如果没有W开头的信息就代表安装成功。

注:如果想要安装conda没有而pip有的包,可以使用如下方式安装到当前虚拟环境下:

python -m pip install numpy
  • 2
    点赞
  • 22
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值