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

系统 mint18  x64

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

2安装Anaconda3

https://www.cnblogs.com/xuanmanstein/p/13892149.html

 

3Pytorch

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

ubuntu 安装cuda 9.1 pytorch 0.3.0

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

 ubuntu 安装cuda 9.1 pytorch 0.3.0

结果还是极其慢,我倒。

只好从源码编译了

参考这个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)
虽然这一步还是有点慢,但是已经可以忍受了

ubuntu 安装cuda 9.1 pytorch 0.3.0

 还是特别慢。

 

最后参考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

相关文章: