ubuntu16.04 配置环境(tensorflow、pytorch)的注意事项

PYTHON

首先,python 在ubuntu16.04中是默认安装python2.7以及python3.5的  可以输入pytho -v 以及python3进行查看。

其次,其默认环境是python2.7,现需要更改默认版本为3.5

shell里执行:

sudo update-alternatives --install /usr/bin/python python /usr/bin/python2 100
sudo update-alternatives --install /usr/bin/python python /usr/bin/python3 150

效果如下:

此时,在命令端输入安装命令则会自动安装支持py3的包。

Numpy:

sudo apt-get install python3-numpy

Scipy:

sudo apt-get install python3-scipy

Matplotlib:

sudo apt-get install python3-matplotlib

Pandas:

sudo apt-get install python3-pandas

统计建模分析工具statsModels:

sudo apt-get install python3-statsmodels

机器学习库scikit-learn:

sudo apt-get install python3-sklearn

xgboost:

sudo -H pip3 install --pre xgboost

 seaborn,作图工具,是matplotlib的补充,制作更具吸引力的图:

sudo pip3 install seaborn

 

如果要切换到Python2,执行:

sudo update-alternatives --config python3

按照提示输入选择数字回车即可。系统会自动选择下载。

Tensorflow

GPU

Tensorflow 已经不再支持 mac 的 GPU 版了, 下面是 Linux 安装 GPU 版的说明. 说先安装 NVIDIA CUDA 必要组建.

$ sudo apt-get install libcupti-dev

然后确保你的 linux 上 pip 是可用的, 接着我们可以直接通过pip 安装:

$ sudo apt-get install python-pip python-dev   # for Python 2.7
$ sudo apt-get install python3-pip python3-dev # for Python 3.n

然后选择你想要cpu 或者 gpu 版本.

$ pip install tensorflow      # Python 2.7; CPU support (no GPU support)
$ pip3 install tensorflow     # Python 3.n; CPU support (no GPU support)
$ pip install tensorflow-gpu  # Python 2.7;  GPU support
$ pip3 install tensorflow-gpu # Python 3.n; GPU support

$pip3 install pytorch-gpu  #Python 3.n support

根据莫烦教程的安装代码后发现后来不能够使用GPU版本,所以发现还是要安装CUDA和CUDNN,我选择的版本是cuda8.0,cudnn 5.0的版本.

然后重新安装了tensorflow 和Pytorch

安装完毕后可以使用,当然后期会遇到一些包是没有安装的,

所以直接sudo pip3 install ***(包)   就可以了




  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值