安装GPU版本MXNet

直接安装方法

sudo apt-get update
sudo apt-get install -y python3 python3-pip
确认当前计算机CUDA版本（nvcc --version），如果没有CUDA，则参考cuda安装及版本切换方法，cudnn安装方法
安装MXNet：pip3 install mxnet-cu80
安装指定版本的MXNet：pip3 install mxnet-cu80==1.3.1
测试是否安装成功：
- 进入python环境
- import mxnet as mx
- a = mx.nd.array([1,2,3]).as_in_context(mx.gpu(0))
- b = mx.nd.array([4,5,6]).as_in_context(mx.gpu(0))
- print(a+b)

源码安装方法（参考https://zhuanlan.zhihu.com/p/165275286）

源码下载：https://mxnet.apache.org/versions/1.7.0/get_started/download.html（以apache-mxnet-src-1.5.1-incubating.tar.gz版本为例）
安装一些依赖包：(可能不全，在源码编译mxnet时需要啥再安装啥）
- sudo apt-get groupinstall "Development Tools"
- sudo apt-get install atlas-devel opencv-devel
- sudo apt-get install git make ninja-build automake autoconf libtool protobuf-compiler protobuf-devel atlas-devel openblas-devel lapack-devel openssl-devel zeromq-devel
mxnet源码编译（整个过程大概20多分钟）
- tar -zxvf apache-mxnet-src-1.5.1-incubating.tar.gz
- cd apache-mxnet-src-1.5.1-incubating
- make -j$(nproc)
安装验证
- find . -name "libmxnet.so" # 能找到证明安装是对的
- ldd lib/libmxnet.so # 这一步一定要有

Docker镜像安装方法

sudo docker pull mxnet/python:1.3.1_gpu_cu80_py3
其他MXNet镜像可以去MXNet镜像库搜索

卸载MXNet方法

pip3 uninstall mxnet-cu80==1.3.1
sudo docker rmi mxnet/python:1.3.1_gpu_cu80_py3

安装GPU版本MXNet相关推荐

Win10 64位下GPU版本MXNet+Tensorflow 1.3.0的安装
说明:本文记录时间为2017.7.23.在此时Tensorflow在Win下的安装要求为系统64位且Python版本为3.5(现已经支持py 3.6),MXNet在Win下只支持Python 2. 更 ...
Win10 64bit +VS2015+GPU版本MXNet+Tensorflow 1.3.0的安装
注,尊重原创,原文出自:http://blog.csdn.net/u010469126/article/details/75891967 其中蓝色部分为改动的() 说明:本文记录时间为2017.7.2 ...
用MXnet实战深度学习之一：安装GPU版mxnet并跑一个MNIST手写数字识别（zz）
用MXnet实战深度学习之一:安装GPU版mxnet并跑一个MNIST手写数字识别我想写一系列深度学习的简单实战教程,用mxnet做实现平台的实例代码简单讲解深度学习常用的一些技术方向和实战样例.这 ...
tensorflow运行环境linux,在ubuntu或者min运行环境下安装gpu版本的tensorflow
转载请注明出处:blog.csdn.net/sproll 本文描述在ubuntu系列操作系统上安装gpu版本的tensorflow的过程. 0,准备工作 BIOS中关闭板载显卡,显示器接在nvidia ...
英伟达3060Ti安装GPU版本TensorFlow2.X
查看Python与TensorFlow对应版本安装GPU版本的TensorFlow的时候,我们需要考虑的一个问题是Python版本与TensorFlow版本的对应关系,可以参考下面这个链接: Pyt ...
windows10 安装gpu版本TensorFlow脑壳疼自我记录
写文章时间为:2018.10.26 显卡为NVIDIA GeForce RTX 2080Ti(刚出没有多久),系统为window10 放在开始想说的是,以下是我安装过程,仅是自我总结,仅供参考. 再之 ...
常识——windows的tensorflow安装gpu版本,cuda算力只有3.0也可以
文章目录问题 windows安装tensorflow_gpu版本检查显卡是否支持cuda 查看自己的显卡型号去官网查看是否支持cuda 检查显卡的cuda算力按照GPU算力3.0的电脑配置进行 ...
安装 GPU 版本的 tensorflow 完整版本
前言: 之前安装的 CPU 版本的 tensorflow 一直出问题,索性就直接安装 GPU 版本的 tensorflow 了(有了GPU 就不能浪费). 安装过程: 1)看自己有无 GPU,找到对应 ...
pytorch安装GPU版本 (Cuda12.1)教程
使用本教程前,默认您已经安装并配置好了python3以上版本 1. 去官网下载匹配的Cuda Cuda下载地址当前最高版本的Cuda是12.1 我安装的就是这个版本小提示:自定义安装可以只选择安装 ...

安装GPU版本MXNet

直接安装方法

源码安装方法（参考https://zhuanlan.zhihu.com/p/165275286）

Docker镜像安装方法

卸载MXNet方法

安装GPU版本MXNet相关推荐

最新文章

热门文章