一、安装anaconda

1.安装anaconda

anaconda官网下载.sh安装包

在下载路径下安装anaconda

bash Anaconda3-2020.07-Linux-x86_64.sh

一路回车直到,是否接受协议?yes
安装目录选择默认目录则直接回车,若自定义目录则输入路径。
是否初始化?yes (后面可以省去conda命令不生效,修改bashrc的步骤)

2.配置路径

安装完成后默认配置路径需要输入如下命令行

source ~/.bashrc

关掉终端窗口并重启,输入conda list命令验证

二、安装pytorch

1.创建pytorch虚拟环境

使用anaconda安装tensorflow,caffe、pytorch等框架,可以先通过anaconda创建虚拟环境,然后在不同的虚拟环境下装上不同的版本的框架(例如在一个虚拟环境上安装一个tensorflow1.4,在另一个虚拟环境中上安装一个tensorflow,在另一个虚拟环境中安装pytorch)。并且如果配置出错,直接删除虚拟环境的包即可,而不影响这个系统。
代码如下:

 conda create -n pytorch python=3.7

运行结果
在这里插入图片描述

2.激活pytorch虚拟环境

 source activate pytorch

激活后,会看到前面会显示环境名字比如(pytorch),如下
在这里插入图片描述

3.在虚拟环境下安装pytorch

打开pytorch起始页,根据所使用Linux服务器的信息,进行选择:
在这里插入图片描述
选择完后,在命令行执行上图红中的命令即可

 conda install pytorch torchvision torchaudio cudatoolkit=11.0 -c pytorch

在这里插入图片描述

4.验证是否安装成功

执行下面命令行验证

python  # 进入python 环境
import torch  # 导入安装的pytorch包
torch.cuda.is_available()  # 检查cuda是否可以使用

输出结果
在这里插入图片描述
输出True即为成功。

Logo

汇聚全球AI编程工具,助力开发者即刻编程。

更多推荐