虚拟机Ubuntu20安装tensorflow-rocm

AMD 的显卡才需要安装TensorFlow-rocm 英伟达的不需要看本篇博客咯。

首先我们先安装OpenBLAS, OpenBLAS的作用是AMD CPU的numpy计算加速器和mkl是因特尔的numpy计算加速器的作用一样，所以英特尔CPU可以安装mkl

# 下载
sudo apt install git
git clone https://github.com/xianyi/OpenBLAS.git# 安装
cd OpenBLAS/
sudo make TARGET=NEHALEM
sudo make PREFIX=/opt/OpenBLAS install# 测试
vim main.c

内容如下

#include <cblas.h>
#include <stdio.h>void main() {int i = 0;double A[6] = {1.0,2.0,1.0,-3.0,4.0,-1.0};double B[6] = {1.0,2.0,1.0,-3.0,4.0,-1.0};double C[9] = {.5,.5,.5,.5,.5,.5,.5,.5,.5};int M = 3; // row of A and Cint N = 3; // col of B and Cint K = 2; // col of A and row of Bdouble alpha = 1.0;double beta = 0.0;cblas_dgemm(CblasRowMajor, CblasNoTrans, CblasNoTrans, M, N, K, alpha, A, K, B, N, beta, C, N);for (i = 0; i < 9; i++) {printf("%lf ", C[i]);}printf("\n");
}

之后执行编译和运行

gcc main.c -o main -I /opt/OpenBLAS/include/ -L /opt/OpenBLAS/lib/ -lopenblas -lpthread
./main
# 当报错找不到libopenblas.a时 执行以下命令 =_+...
sudo cp /opt/OpenBLAS/lib/libopenblas.* /usr/lib/
./main
# 完成安装！

之后我们下载安装AMDGPU驱动

驱动安装页
https://www.amd.com/zh-hans/support/kb/release-notes/rn-amdgpu-unified-linux-20-20

# 安装GPU驱动
tar -Jxf amdgpu-pro-20.20-1098277-ubuntu-20.04.tar.xz
cd amdgpu-pro-20.20-1098277-ubuntu-20.04/
./amdgpu-install# 驱动安装完成后测试
/opt/rocm/bin/rocminfo
/opt/rocm/opencl/bin/clinfo
# 若仅有一个成功识别显卡，仍可以尝试使用tensorflow-rocm。# 安装apt的rocm的key和包列表
sudo apt update
sudo apt dist-upgrade
sudo apt install libnuma-dev
wget -q -O - http://repo.radeon.com/rocm/rocm.gpg.key | sudo apt-key add -
echo 'deb [arch=amd64] http://repo.radeon.com/rocm/apt/debian/ xenial main' | sudo tee /etc/apt/sources.list.d/rocm.list# 正式安装rocm-dkms
sudo apt update
sudo apt install rocm-dkms

当出现以下错误时不要惊慌，不用管它。

Loading new amdgpu-4.0-23 DKMS files...
Building for 5.8.0-45-generic
Building for architecture x86_64
Building initial module for 5.8.0-45-generic
ERROR: Cannot create report: [Errno 17] File exists: '/var/crash/rock-dkms.0.crash'
Error! Bad return status for module build on kernel: 5.8.0-45-generic (x86_64)
Consult /var/lib/dkms/amdgpu/4.0-23/build/make.log for more information.
dpkg: 处理软件包 rock-dkms (--configure)时出错：已安装 rock-dkms 软件包 post-installation 脚本 子进程返回错误状态 10
dpkg: 依赖关系问题使得 rocm-dkms 的配置工作不能继续：rocm-dkms 依赖于 rock-dkms；然而：软件包 rock-dkms 尚未配置。dpkg: 处理软件包 rocm-dkms (--configure)时出错：依赖关系问题 - 仍未被配置
因为错误消息指示这是由于上一个问题导致的错误，没有写入 apport 报告。在处理时有错误发生：rock-dkmsrocm-dkms
E: Sub-process /usr/bin/dpkg returned an error code (1)

最后还需要安装一个包

sudo apt install rccl# 查看rocm版本
apt show rocm-libs -a
apt show rocm-dkms -a

# 最后安装TensorFlow-rocm
pip3 install tensorflow-rocm -i https://pypi.tuna.tsinghua.edu.cn/simple# 测试
python3
import tensorflow
# 不报错则安装成功！

虚拟机Ubuntu20安装tensorflow-rocm相关推荐

从0开始ubuntu16.04（虚拟机）安装tensorflow，spinningup
下载ubuntu16.04镜像ISO文件,然后创建虚拟机. 一.基础环境配置 (1)查看文件工具安装vim (2)虚拟机文件传输工具(可自行选择) 安装ssh 用xftp传输文件,如下图二.配置开 ...
虚拟机VMware的Ubuntu下安装tensorflow详解
这里笔者已经安装好了虚拟机并安装了Ubuntu系统,并且虚拟机可以联网,因此就直接开始介绍下面的步骤.网上博文很乱,特意整理,以供参考. 虚拟机VMware的Ubuntu下安装tensorflow详解 ...
虚拟机下安装anaconda3+python3.6+TensorFlow
(小白)虚拟机下安装anaconda3+python3.6+TensorFlow 一.安装Anaconda 1.到https://www.anaconda.com/download/下载对应版本的an ...
在Win10 Anaconda中安装Tensorflow
离完成上一篇文章有近1年了.2016年发生了太多的事情,从而没能坚持哪怕是每月一篇这样的频率.终于在2017年的1月份抽出几天搞出了一些东西.一路坑洼,赶紧记录下来. 2016年初就开始看深度学习的东 ...
ubuntu-Linux下如何安装Tensorflow？
ubuntu下如何安装Tensorflow? AI这个概念好像突然就火起来了,年初大比分战胜李世石的AlphaGo成功的吸引了大量的关注,但其实看看你的手机上的语音助手,相机上的人脸识别,今日头条上帮 ...
Ubuntu 15.04 安装TensorFlow（源码编译）及测试梵高作画
介绍Google的TensorFlow机器学习开源库,在UbuntuKylin上的安装和和源码编译. 原始官方文档参见:http://www.tensorflow.org. 本电脑配置如下: 3.19 ...
Ubuntu安装tensorflow报错：tensorflow-xx.whl not a supported wheel on this platform
解决Ubuntu安装tensorflow报错:tensorflow-0.5.0-cp27-none-linux_x86_64.whl is not a supported wheel on this ...
windows下安装TensorFlow（Win8 + Anaconda4.4 python3.6）
本文主要介绍如何在Windows下安装TensorFlow.当然建议还是在linux下安装,在windows下开发不太方便. 相关背景网上现有的方法都是教如何在windows下安装虚拟机 ...
linux虚拟机 Ubuntu 安装方法
1.1 下载 VMware Workstation 软件 https://www.vmware.com/cn/products/workstation-pro/workstation-pro-eval ...

虚拟机Ubuntu20安装tensorflow-rocm

虚拟机Ubuntu20安装tensorflow-rocm相关推荐

最新文章

热门文章