文章目录
1.安装Anaconda
主要参考: Ubuntu18.04 安装 Anaconda3
1.1.下载合适版本的Anaconda安装包
下载 Anaconda3-5.2.0-Linux-x86_64.sh (Ubuntu18.04的对应的Anaconda版本),注意因为Ubuntu18的发行版带的是python3.6,而适配python3.6的Anaconda版本就是上面这个版本。
如果直接到Anaconda官网去下载的话,需要去找过去的版本,并且网速也不快。所以一个比较好的方法是到清华的镜像网站去下载。
1.2.安装Anaconda
- 进入下载目录,一般是~/Downloads
cd Downloads/
- 运行.sh文件
如果是bash终端,则运行
bash Anaconda3-5.2.0-Linux-x86_64.sh
如果是zsh终端,则运行
zsh Anaconda3-5.2.0-Linux-x86_64.sh
- 进入注册信息页面,输入yes
- 阅读注册信息,然后输入yes;查看文件即将安装的位置,按enter,即可安装
- 安装完成后,收到加入环境变量的提示信息,输入yes
注意:这上面添加环境变量是默认添加到.bashrc这个文件中的,也就是添加到bash中。如果使用的终端是zsh,那么要在安装完成之后手动添加环境变量到.zshrc中,在.zshrc文件中添加下面这句话:
# added by Anaconda3 installer
export PATH="/home/cc/anaconda3/bin:$PATH"
- 提示信息“Do you wish to proceed with the installation of Microsoft VSCode? [yes|no]”,输入no
- 安装完毕,测试安装是否完成
终端输入conda --version
,如果能够正常显示版本,那么说明安装成功。
1.3.创建新环境
conda create -n 环境名 依赖条件
,比如创建一个名为pytorch
的环境,依赖的是python3.6
的环境,输入以下语句即可。
conda create -n pytorch python=3.6
1.4.查看环境
-
conda env list
查看虚拟环境,如下运行结果是我配置了另一个环境之后的结果,其中base是本来就有的环境,pytorch是我配置的另一个虚拟环境 -
source activate 环境名
激活环境,也就是切换到另一个环境中 -
conda list
查看当前环境中安装的包,如下图是我切换到自己配置pytorch环境中安装的包
2.安装pytorch
2.1.进入pytorch官网,查询安装命令
- 版本选择稳定版
Stable
- Package选择
Conda
(可能选pip也行?) - 计算平台如果有独立GPU的话就选CUDA10.2(注意在安装Anaconda的时候就安装了CUDA了,不用再单独安装了)。如果没有GPU那么就选择CPU
- 复制官网生成的命令,然后先激活自己新建的虚拟环境,然后在这个虚拟环境里运行官网生成的命令安装pytorch
2.2.测试pytorch安装是否成功
- 使用命令查看torch版本
python
import torch
torch.__version__
- 使用命令查看是否可加速,如果输出True说明可以使用GPU进行CUDA加速
print(torch.cuda.is_available())
- 使用一个矩阵运算测试是否能正常运行:
import torch as t
x = t.rand(5,3)
y = t.rand(5,3)
x = x.cuda()
y = y.cuda()
print(x+y)
测试结果:
2.3.补充安装一些包
参考:解决jupyter notebook无法找到虚拟环境的问题
在上面安装pytorch的过程中,会安装很多包。但是如果安装之后直接输入jupyter notebook
打开jupyter,会发现右上角的环境里没有自己设置的虚拟环境。这个是因为还缺两个包不能用。
- 首先进入到自己创建的虚拟环境(pytorch是虚拟环境的名字)
source activate pytorch
- 在该环境下下载如下两个库
conda install ipykernel
conda install nb_conda
注意:如果是在windows上进行安装的anaconda,在这一步使用conda安装其他的包的时候,可能会报错“无法定位程序输入点XXX于XXX.dll上”,如下图所示。解决办法参考 解决conda安装包时出现“无法定位程序输入点XXX于XXX.dll上”的问题
但是实际上我测试了在windows下即使执行了上面的操作,后面再使用conda install的时候还是会报错,并且进入jupyter notebook之后运行一些命令的时候也会报错,最后直接把这个版本的anaconda卸载了,安装了一个比较新的版本的windows下的anaconda,然后就不报错了。
- 打开jupyter notebook,我们就可以看到我们创建的虚拟环境pytorch了
在命令行输入jupyter notebook
- 同样,在jupyter中也可以使用上面测试pytorch的语句进行测试pytorch环境是否安装成功