在Ubuntu17.10搭建pytorch环境

来源:互联网 发布:amtemu v0.8.1 mac 编辑:程序博客网 时间:2024/05/21 22:57

(1)Ubuntu17.10已经自带了python3.6.3,我们需要先下载一个对应版本的anaconda。

在终端进入anaconda的目录后,调用命令dash Anaconda3-5.0.1-Linux-x86_64.sh安装anaconda。

安装过程中会有提示,让你按enter确定是否安装在当前位置之类的,最后有一个很重要的提示,提示你是否将anaconda加入环境变量选择yes。

我当时不知道由于操作比较快,enter按太多的原因,我没有来得及敲yes,系统就自动选择了no。。。。既然如此只能手动安装了。

一种可以修改当前用户的环境变量的方法。

在终端输入:

gedit.bashrc打开.bashrc文件

在最末端加上:

export PATH="/home/zhang/anaconda3/bin:$PATH"
export LD_LIBRARY_PATH=/home/zhang/anaconda3/lib:$LD_LIBRARY_PATH

(之前有帖子让我加的是:export CMAKE_PREFIX_PATH=/home/zhang/anaconda3/bin,让我搞了半天,还为之后装opencv埋一下了一个深坑。。。。)

保存并退出。

source ~/.bashrc

然后再输入import conda,发现就可以找到conda命令了。


(2)安装依赖库

conda install numpy mkl setuptools cmake gcc cffi

conda install -c soumith magma-cuda7.5

如果没有cuda,后面条命令就写:

conda install -c soumith

(3)安装pytorch

conda install -c peterjc123 pytorch=0.1.12

最后在python命令框里输入:import torch 没有问题就成功了