linux GitHub安装 Pytorch

2017-10-22  本文已影响0人  zhaokuanyew

前言

因为网络原因(你懂的),用官网的安装办法根本无法安装pytorch,无奈去github看到了源码安装。话不多说,开始安装吧

访问github

pytorch在github上的地址

pytorch有源码安装的教程:


官网的教程

github推荐了2种安装方法

这里就介绍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安装成功测试
上一篇下一篇

猜你喜欢

热点阅读