linux GitHub安装 Pytorch
2017-10-22 本文已影响0人
zhaokuanyew
前言
因为网络原因(你懂的),用官网的安装办法根本无法安装pytorch,无奈去github看到了源码安装。话不多说,开始安装吧
访问github
pytorch有源码安装的教程:
官网的教程
github推荐了2种安装方法
- 官网上的3种方法(conda,pip等)
- 源码安装
这里就介绍Install From Source
准备工作-安装anaconda
官网也强力推荐anaconda,这个可以作为python版本管理工具,真心非常好用,而且集成各种第三方库。
If you are installing from source, we highly recommend installing an Anaconda environment. You will get a high-quality BLAS library (MKL) and you get a controlled compiler version regardless of your Linux distro.
但是还是那个原因,下载太慢了,可以从我的百度云下载
百度云链接 密码:lotq
其中有个要注意的,就是anaconda的配置。就是加入你错过了点击yes的机会,anaconda就不会帮你配置好环境变量了。所以要进行以下操作:
# 将anaconda的bin目录加入PATH
echo 'export PATH="~/anaconda3/bin:$PATH"' >> ~/.bashrc
# 更新bashrc以立即生效
source ~/.bashrc
最后确认一下,如图
anaconda安装成功效果
准备工作-安装前提
export CMAKE_PREFIX_PATH="$(dirname $(which conda))/../" # [anaconda root directory]
echo 'export PATH="~/anaconda3/bin"' >> ~/.bashrc
# 或者粘贴下一行到.bashrc中
export PATH=~/anaconda3/bin
# 然后
source ~/.bashrc
# 安装基本的依赖
conda install numpy pyyaml mkl setuptools cmake gcc cffi
# Add LAPACK support for the GPU
conda install -c soumith magma-cuda80 # or magma-cuda75 if CUDA 7.5
(其实这个也挺慢的)
安装-clone并安装
安装成功后,就把项目clone下来吧
git clone --recursive https://github.com/pytorch/pytorch
然后
python setup.py install
终于安装好了,挺麻烦的,最好还是科学上网,然后用官网的方法安装吧
pytorch安装成功测试