mxnet/tensorflow/pytorch 安装

简介

mxnet 官方文档

tensorflow 官方文档

pytorch 官方文档

安装

mxnet 对应版本

tensorflow linux 对应版本

pytorch 对应版本

在安装mxnet/tensorflow/pytorch 一定要注意cuda版本的对应关系

1
2
3
4
5
6
7
8
9
10
11
12

# 安装mxnet 默认是 CPU
pip3 install mxnet

## ubuntu 安装mxnet GPU
pip3 install mxnet-cu90

# 安装 pytorch GPU
pip3 install torch torchvision

# 安装 tensorflow GPU
pip install tensorflow_gpu==1.12.0
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
# 验证 torch GPU

import torch
torch.cuda.is_available()

# 验证 mxnet GPU

import mxnet as mx
from mxnet import nd
from mxnet.gluon import nn
a = nd.array([1, 2, 3], ctx=mx.gpu())
printa(a)


# 验证 tensorflow gpu

import tensorflow as tf

with tf.device('/cpu:0'):
    a = tf.constant([1.0,2.0,3.0],shape=[3],name='a')
    b = tf.constant([1.0,2.0,3.0],shape=[3],name='b')
with tf.device('/gpu:1'):
    c = a+b

sess = tf.Session(config=tf.ConfigProto(allow_soft_placement=True,log_device_placement=True))
#sess = tf.Session(config=tf.ConfigProto(log_device_placement=True))
sess.run(tf.global_variables_initializer())
print(sess.run(c))