ubuntu 安装cuda 9.1 pytorch 0.3.0

毕业再没用配过机器学习的环境了，既亲切又陌生，久违了。

系统 mint18  x64

1安装cuda

  按官网提示 选的9.1版  https://developer.nvidia.com/cuda-toolkit

2安装Anaconda3

  从清华的镜像站下载https://mirrors.tuna.tsinghua.edu.cn/anaconda/archive/

  从最下面选Anaconda3-5.0.1-Linux-x86_64

 在下载 路径下 执行 

   bash ./Anaconda3-5.0.1-Linux-x86_64.sh  

安装好之后，按提示

export PATH=/home/machinelearning/anaconda3/bin:$PATH

并添加到

xed ~/.profile  （不是bashrc，因为换成了zsh）

然后source ~/.profile

切换清华的源

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
conda config --set show_channel_urls yes

安装好之后，已经可以使用

anaconda-navigator

3Pytorch

直接在官网按按钮选版本http://pytorch.org/

如果没切换清华源，可能会奇慢无比。

 

结果还是极其慢，我倒。

只好从源码编译了

参考这个http://blog.csdn.net/draco_mystack/article/details/71191924 和pytorch的github https://github.com/pytorch/pytorch#from-source

CMAKE_PREFIX_PATH="$(dirname $(which conda))/../"

conda install numpy pyyaml mkl setuptools cmake cffi  
（还好，这几个清华的镜像站里都有）

conda install -c soumith magma-cuda90
（我的cuda下载nv官网的9.1 参考pytorch官网生成的安装命令行，所以弄了个90.原文是80）

虽然这一步还是有点慢，但是已经可以忍受了

 还是特别慢。

最后参考https://github.com/dnouri/skorch

直接从anaconda里起控制台，pip安装了

pip install http://download.pytorch.org/whl/cu90/torch-0.3.0.post4-cp36-cp36m-linux_x86_64.whl