cpu、gpu 安装框架pytorch，cntk,theano及测试

时间：2021-06-21 19:53:57

一，cpu 下安装

tensorflow

conda env list

source activate tensorflow

直接安装相应版本

python

import tensorflow as tf

tf.version 1.11.0

keras 直接安装

conda env list

source activate keras

import keras 2.2.2

print(keras.version)

import tensorflow as tf

tf.version

pytorch

import torch

print(torch.version)

print(torch.cuda.device_count())

print(torch.cuda.is_available())

cntk

/root/anaconda3/bin/conda env list

source activate cntk-py35

python 3.5.6

export PATH=/root/anaconda3/bin:$PATH

python -c "import cntk; print(cntk.version)"

theano

caffe2

python 3.6.9

import caffe2

安装

conda create -n caffe2 python=3.6

conda activate caffe2

conda install pytorch-nightly-cpu -c pytorch -n caffe2

python -c 'from caffe2.python import core' 2>/dev/null && echo "Success" || echo "Failure"

报错:

pip install protobuf

pip install future

参考官网安装即可

gpu

tensorflow-gpu:1.11.0 python 3.5

export PATH=/root/anaconda3/bin:$PATH

source activate tensorflow

keras

export PATH=/root/anaconda3/bin:$PATH

conda env list

source activate keras

python3.5

nvidia-docker run -it --rm pytorch-gpu:1.1.0 /bin/bash

pytorch

[root@191ddd30d4ae /]# python

Python 3.6.9 |Anaconda, Inc.| (default, Jul 30 2019, 19:07:31)

[GCC 7.3.0] on linux

Type "help", "copyright", "credits" or "license" for more information.

import torch

print(torch.version)

1.1.0

print(torch.cuda.device_count())

1

print(torch.cuda.is_available())

True

cntk

source activate cntk-py35 python3.5

python -c "import cntk; print(cntk.version)"

2.4

theano

gpu-theano-in-use:1.0.4 python2.7

source activate theano

python test.py

import theano

/root/anaconda3/envs/theano/lib/python2.7/site-packages/theano/gpuarray/dnn.py:184: UserWarning: Your cuDNN version is more recent than Theano. If you encounter problems, try updating Theano or downgrading cuDNN to a version >= v5 and <= v7.

warnings.warn("Your cuDNN version is more recent than "

Using cuDNN version 7603 on context None

Mapped name None to device cuda: GeForce GTX 960M (0000:01:00.0)

theano.version

u'1.0.4'

https://www.jianshu.com/p/4cc75a79dce9

Linux下安装miniconda

在官网下载miniconda3

执行：bash Miniconda3-latest-Linux-x86_64.sh 　

－vim ~/.bashrc

－export PATH=~/anaconda3/bin:$PATH

－source ~/.bashrc

创建虚拟环境并安装theano

基于python2.7创建一个名为theano的环境

conda create --name theano python=2.7

进入虚拟环境： source activate theano

－使用conda安装：conda install numpy scipy mkl

pip install parameterized

conda install theano pygpu

       －使用pip安装：pip install Theano

测试参考官网文档

caffe2

看官网文档安装

https://caffe2.ai/docs/getting-started.html?platform=ubuntu&configuration=compile

https://blog.csdn.net/qq_35451572/article/details/79428167

cmake

-DCUDA_TOOLKIT_ROOT_DIR=/usr/local/cuda-9.0

-DCUDNN_ROOT_DIR=/usr/local/cuda

To check if Caffe2 build was successful

python -c 'from caffe2.python import core' 2>/dev/null && echo "Success" || echo "Failure"

To check if Caffe2 GPU build was successful

This must print a number > 0 in order to use Detectron

python -c 'from caffe2.python import workspace; print(workspace.NumCudaDevices())'

参考

https://blog.csdn.net/Yan_Joy/article/details/70241319

https://www.nvidia.com/en-gb/data-center/gpu-accelerated-applications/caffe2/

https://blog.csdn.net/qq_35451572/article/details/79428167

https://blog.csdn.net/qq_16525279/article/details/79724728

https://blog.csdn.net/y_f_raquelle/article/details/83278953

https://www.cnblogs.com/nanzhao/p/9596844.html

附:conda常用

  1. conda env list 或 conda info -e 查看当前存在哪些虚拟环境

  2. conda update conda 检查更新当前conda

  3. conda update --all 更新本地已安装的包

  4. conda create -n your_env_name python=X.X（2.7、3.6等） anaconda 命令创建python版本为X.X、名字为your_env_name的虚拟环境。your_env_name文件可以在Anaconda安装目录envs文件下找到。

  5. Windows: activate your_env_name(虚拟环境名称) 激活虚拟环境

  6. conda install -n your_env_name [package] 安装package到your_env_name中

  7. linux: source deactivate Windows: deactivate 关闭虚拟环境

  8. conda remove -n your_env_name(虚拟环境名称) --all 删除虚拟环境

  9. conda remove --name your_env_name package_name 删除环境中的某个

相关文章