最近在学习pytorch需要在服务器搭建一个安装pytorch框架,所以结合我这几天的学习经历写了一个在Ubuntu16.04版环境下安装pytorch。
1.安装anaconda
anaconda是一个很好用的python包管理器,对于经常使用python是非常友好的,可以在anaconda下新建不同的环境,以满足不同开发环境的需求。anaconda的安装这个网上有很多教程,不做具体的的描述。
2.建一个anaconda环境
- 这是新建一个名字为pytorchgpu的环境,python版本为3.6,可根据需要更改
conda create -n pytorchgpu python=3.6
- 激活环境
source activate pytorchgpu
- 如果想要退出这个环境,可以使用下面的命令
source deactivate pytorchgpu
3.修改conda的下载源
pytorch最近已经更新到了稳定版本的1.7,从Pytorch官网上可以看到有多种安装方式:
比较常见的方法是通过pip和conda安装,当使用conda安装的时候,可能会遇到下载过慢的问题,尤其是文件:pytorch包文件,下载的时候可能会遇到无尽的等待,甚至失败(亲身体会)。这里推荐用清华源替代默认conda源的方法,可以解决下载过慢的问题。因此需要添加国内的下载源,我使用的清华的镜像:
清华conda源地址:https://mirrors.tuna.tsinghua.edu.cn/help/anaconda/
添加清华源
输入下面的命令: